From c0c7ea2db5daddece40b5377afd2f7d9e3c79147 Mon Sep 17 00:00:00 2001
From: PGijsbers <p.gijsbers@tue.nl>
Date: Sat, 29 Apr 2023 19:37:15 +0200
Subject: [PATCH] Load data as pandas dataframe

Because some pixels were inferred as categorical.
See also #193
---
 tests/system/test_gama.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/system/test_gama.py b/tests/system/test_gama.py
index a8cc86e4..93cedb35 100644
--- 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
     )