Spaces:
Running
Running
MilesCranmer
commited on
Commit
•
faa83d3
1
Parent(s):
0bf77e2
Fix broken test statement
Browse files- test/test.py +1 -1
test/test.py
CHANGED
@@ -90,7 +90,7 @@ class TestPipeline(unittest.TestCase):
|
|
90 |
regressor.fit(X, y)
|
91 |
|
92 |
self.assertLessEqual(regressor.equations.iloc[-1]["MSE"], 1e-4)
|
93 |
-
np.
|
94 |
|
95 |
def test_noisy(self):
|
96 |
|
|
|
90 |
regressor.fit(X, y)
|
91 |
|
92 |
self.assertLessEqual(regressor.equations.iloc[-1]["MSE"], 1e-4)
|
93 |
+
np.testing.assert_almost_equal(regressor.predict(X), y, decimal=1)
|
94 |
|
95 |
def test_noisy(self):
|
96 |
|