add: fix for ag 1.5

Author: LennartPurucker

tabarena/pyproject.toml

@@ -15,7 +15,7 @@ requires-python = ">=3.10"
 #   uv pip install --prerelease=allow .
 dependencies = [
   # TODO: To use `uv`, you need to do `uv pip install --prerelease=allow .` so it recognizes pre-release AutoGluon
-  "autogluon>=1.4.1b20250910,<1.5",  # TODO: Remove after moving `benchmark` code elsewhere
+  "autogluon>=1.4.1b20250910,<1.6",  # TODO: Remove after moving `benchmark` code elsewhere
   "bencheval",
   "openml>=0.14.1",  # consider making optional
   "pyyaml",

tabarena/tabarena/benchmark/models/ag/tabdpt/tabdpt_model.py

@@ -13,8 +13,8 @@
 
 
 class TabDPTModel(AbstractModel):
-    ag_key = "TABDPT"
-    ag_name = "TabDPT"
+    ag_key = "TA-TABDPT"
+    ag_name = "TA-TabDPT"
     seed_name = "seed"
     default_random_seed = 0
 

tabarena/tabarena/benchmark/models/ag/tabpfnv2_5/tabpfnv2_5_model.py

@@ -290,8 +290,8 @@ class RealTabPFNv25Model(TabPFNModel):
     The extra checkpoints include models trained on only synthetic datasets as well.
     """
 
-    ag_key = "REALTABPFN-V2.5"
-    ag_name = "RealTabPFN-v2.5"
+    ag_key = "TA-REALTABPFN-V2.5"
+    ag_name = "TA-RealTabPFN-v2.5"
 
     default_classification_model: str | None = (
         "tabpfn-v2.5-classifier-v2.5_default.ckpt"

tabflow_slurm/submit_template.sh

@@ -1,7 +1,7 @@
 #!/bin/bash
 #SBATCH --ntasks=1
 #SBATCH --job-name=tabarena_run
-#SBATCH --export=ALL
+#SBATCH --export=ALL,TABPFN_DISABLE_TELEMETRY=1
 #SBATCH --requeue
 #SBATCH --propagate=NONE