From 03b2a76a700449fc9560f18cb5aea77e5f727213 Mon Sep 17 00:00:00 2001 From: Michael Mann Date: Fri, 29 Mar 2024 21:50:30 -0400 Subject: [PATCH] Update ml_test.py --- tests/ml_test.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/ml_test.py b/tests/ml_test.py index 7885b33d..c2786926 100644 --- 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])))