Add tabpfn to model class list

SamoraHunter · SamoraHunter · commit 9e9af9f22a11 · 2026-02-11T21:40:18.000Z
diff --git a/ml_grid/pipeline/model_class_list.py b/ml_grid/pipeline/model_class_list.py
@@ -187,6 +187,7 @@ def get_model_class_list(ml_grid_object: pipe) -> List[Any]:
             "H2O_StackedEnsemble_class",
             "H2O_GAM_class",
             "TabTransformerClass",
+            "TabPFNClassifierClass",
         ]
         for model_name in models_to_disable:
             if model_name in model_class_dict:

Original file line number	Diff line number	Diff line change
`@@ -187,6 +187,7 @@ def get_model_class_list(ml_grid_object: pipe) -> List[Any]:`
`187`	`187`	`"H2O_StackedEnsemble_class",`
`188`	`188`	`"H2O_GAM_class",`
`189`	`189`	`"TabTransformerClass",`
	`190`	`+ "TabPFNClassifierClass",`
`190`	`191`	`]`
`191`	`192`	`for model_name in models_to_disable:`
`192`	`193`	`if model_name in model_class_dict:`