Update ml_test.py

jgrss · Mar 30, 2024 · 03b2a76 · 03b2a76
1 parent 17ee429
commit 03b2a76
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/ml_test.py b/tests/ml_test.py
@@ -107,8 +107,8 @@ def test_tree_predict(self):
                     X, Xy, clf = fit(src, tree_pipeline, aoi_poly, col="lc")
                     y1 = predict(src, X, clf)
                     y2 = fit_predict(src, tree_pipeline, aoi_poly, col="lc")
-                    print("##########/n",y1)
-                    print("##########/n",y2)
+                    print("##########/n",y1.values)
+                    print("##########/n",y2.values)
 
         self.assertTrue(np.all(np.isnan(y1.values[0, 0:5, 0])))
         self.assertTrue(np.all(np.isnan(y2.values[0, 0:5, 0])))