Skip to content

fix: tabpfn version #1407

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
artemlunev2000 merged 2 commits into from
Nov 11, 2025
Merged

fix: tabpfn version #1407

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
61499f0
fix: tabpfn version
artemlunev2000 Nov 11, 2025
b8c70c0
add docstring for tabpfn model
artemlunev2000 Nov 11, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
10 changes: 10 additions & 0 deletions fedot/core/operations/evaluation/operation_implementations/models/tabpfn.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -10,6 +10,16 @@


class FedotTabPFNImplementation(ModelImplementation):
"""FEDOT implementation of TabPFN that uses tabpfn model version 2.
To use the newer TabPFN v2.5 model, you must accept the non-commercial usage license
in your Hugging Face account at https://huggingface.co/Prior-Labs/tabpfn_2_5.
After accepting the license, update the `tabpfn` library to the latest version
(currently 6.0.6) and log in to your Hugging Face account, for example using env variable:
`os.environ["HF_TOKEN"] = "your_hf_token"`. The library will then automatically
use the v2.5 model. If authentication with a token fails, you will get an error.

"""

__operation_params = [
'enable_categorical',
'max_samples',
Expand Down
2 changes: 1 addition & 1 deletion other_requirements/extra.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
# DNNs
tensorflow >= 2.8.0; python_version >= '3.8'
torch >= 1.9.0
tabpfn >= 2.0.0
tabpfn >= 2.0.0, <= 2.2.1

# Images
opencv-python >= 4.5.5.64
Expand Down