Added the ability of the conftest fixture to generate numpy.

alteryx · Aug 15, 2022 · 874f81a · 874f81a
1 parent 0364a0f
commit 874f81a
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/evalml/tests/conftest.py b/evalml/tests/conftest.py
@@ -197,6 +197,10 @@ def _get_test_data_from_configuration(
                 y = ww.init_series(y, logical_type="integer_nullable")
         X = X_all[column_names]
 
+        if input_type == "np":
+            X = X.to_numpy()
+            y = y.to_numpy()
+
         if input_type == "ww":
             logical_types = {}
             if "text" in column_names: