Load data as pandas dataframe

Because some pixels were inferred as categorical. See also #193
openml-labs · Apr 29, 2023 · c0c7ea2 · c0c7ea2
1 parent 05c761a
commit c0c7ea2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/system/test_gama.py b/tests/system/test_gama.py
@@ -8,7 +8,7 @@
 
 
 def _gama_on_digits(gama):
-    X, y = load_digits(return_X_y=True)
+    X, y = load_digits(return_X_y=True, as_frame=True)
     X_train, X_test, y_train, y_test = train_test_split(
         X, y, stratify=y, random_state=0
     )