Spaces:
Sleeping
Sleeping
MilesCranmer
commited on
Commit
•
d85c1a5
1
Parent(s):
1858a22
Fix removal of tmp dir in test
Browse files- test/test.py +2 -1
test/test.py
CHANGED
@@ -41,7 +41,7 @@ class TestPipeline(unittest.TestCase):
|
|
41 |
self.assertLessEqual(equations[0].iloc[-1]["MSE"], 1e-4)
|
42 |
self.assertLessEqual(equations[1].iloc[-1]["MSE"], 1e-4)
|
43 |
|
44 |
-
def
|
45 |
y = self.X[:, [0, 1]] ** 2
|
46 |
w = np.random.rand(*y.shape)
|
47 |
w[w < 0.5] = 0.0
|
@@ -61,6 +61,7 @@ class TestPipeline(unittest.TestCase):
|
|
61 |
**self.default_test_kwargs,
|
62 |
procs=0,
|
63 |
temp_equation_file=True,
|
|
|
64 |
)
|
65 |
|
66 |
np.testing.assert_almost_equal(
|
|
|
41 |
self.assertLessEqual(equations[0].iloc[-1]["MSE"], 1e-4)
|
42 |
self.assertLessEqual(equations[1].iloc[-1]["MSE"], 1e-4)
|
43 |
|
44 |
+
def test_multioutput_weighted_with_callable_temp_equation(self):
|
45 |
y = self.X[:, [0, 1]] ** 2
|
46 |
w = np.random.rand(*y.shape)
|
47 |
w[w < 0.5] = 0.0
|
|
|
61 |
**self.default_test_kwargs,
|
62 |
procs=0,
|
63 |
temp_equation_file=True,
|
64 |
+
delete_tempfiles=False,
|
65 |
)
|
66 |
|
67 |
np.testing.assert_almost_equal(
|