Merge branch 'autogluon:master' into revert_change

tonyhoo · web-flow · commit 5ea605f1824e · 2025-03-11T10:14:35.000-07:00
diff --git a/tabular/setup.py b/tabular/setup.py
@@ -49,7 +49,7 @@
         "fastai>=2.3.1,<2.8",  # <{N+1} upper cap, where N is the latest released minor version
     ],
     "tabpfn": [
-        # versions below 0.1.11 are broken
+        # versions below 2.0.2 are broken (or yanked)
         "tabpfn>=0.1.11,<0.2",  # <{N+1} upper cap, where N is the latest released minor version
     ],
     "tabpfnmix": [
@@ -106,7 +106,7 @@
 
 
 test_requires = []
-for test_package in ["tabpfn", "tabpfnmix", "imodels", "vowpalwabbit", "skl2onnx"]:
+for test_package in ["tabpfnmix", "imodels", "vowpalwabbit", "skl2onnx"]:
     test_requires += extras_require[test_package]
 extras_require["tests"] = test_requires
 install_requires = ag.get_dependency_version_ranges(install_requires)
diff --git a/tabular/tests/unittests/models/test_tabpfn.py b/tabular/tests/unittests/models/test_tabpfn.py
@@ -1,8 +1,11 @@
+import pytest
+
 from autogluon.tabular.models.tabpfn.tabpfn_model import TabPFNModel
 from autogluon.tabular.testing import FitHelper
 
 
 def test_tabpfn_binary():
+    pytest.skip("TabPFN install is broken, skipping test")
     fit_args = dict(
         hyperparameters={TabPFNModel: {}},
     )
@@ -11,6 +14,7 @@ def test_tabpfn_binary():
 
 
 def test_tabpfn_multiclass():
+    pytest.skip("TabPFN install is broken, skipping test")
     fit_args = dict(
         hyperparameters={TabPFNModel: {}},
     )
