Spaces:
Running
Running
MilesCranmer
commited on
Merge pull request #142 from MilesCranmer/deepsource-transform-cf88d7ea
Browse files- test/test.py +1 -2
test/test.py
CHANGED
@@ -107,7 +107,7 @@ class TestPipeline(unittest.TestCase):
|
|
107 |
print("Error in test_multioutput_weighted_with_callable_temp_equation")
|
108 |
print("Model equations: ", model.sympy()[0])
|
109 |
print("True equation: x0^2")
|
110 |
-
|
111 |
try:
|
112 |
np.testing.assert_almost_equal(
|
113 |
model.predict(X.copy())[:, 1], X[:, 1] ** 2, decimal=4
|
@@ -320,4 +320,3 @@ class TestMiscellaneous(unittest.TestCase):
|
|
320 |
with self.assertRaises(Exception) as context:
|
321 |
model.fit(X, y)
|
322 |
self.assertIn("with 10 features or more", str(context.exception))
|
323 |
-
|
|
|
107 |
print("Error in test_multioutput_weighted_with_callable_temp_equation")
|
108 |
print("Model equations: ", model.sympy()[0])
|
109 |
print("True equation: x0^2")
|
110 |
+
|
111 |
try:
|
112 |
np.testing.assert_almost_equal(
|
113 |
model.predict(X.copy())[:, 1], X[:, 1] ** 2, decimal=4
|
|
|
320 |
with self.assertRaises(Exception) as context:
|
321 |
model.fit(X, y)
|
322 |
self.assertIn("with 10 features or more", str(context.exception))
|
|