another fix

ethanglaser · ethanglaser · commit 3f6da9685518 · 2026-02-12T09:44:56.000-08:00
diff --git a/sklbench/datasets/downloaders.py b/sklbench/datasets/downloaders.py
@@ -97,7 +97,7 @@ def fetch_and_correct_openml(
 
     # Get the data with target column specified
     x, y, _, _ = dataset.get_data(
-        dataset_format="dataframe" if as_frame == "auto" or as_frame else "array",
+        dataset_format="dataframe" if as_frame is True else "array",
         target=dataset.default_target_attribute,
     )
 

Original file line number	Diff line number	Diff line change
`@@ -97,7 +97,7 @@ def fetch_and_correct_openml(`
`97`	`97`
`98`	`98`	`# Get the data with target column specified`
`99`	`99`	`x, y, _, _ = dataset.get_data(`
`100`		`- dataset_format="dataframe" if as_frame == "auto" or as_frame else "array",`
	`100`	`+ dataset_format="dataframe" if as_frame is True else "array",`
`101`	`101`	`target=dataset.default_target_attribute,`
`102`	`102`	`)`
`103`	`103`