MilesCranmer commited on
Commit
d85c1a5
1 Parent(s): 1858a22

Fix removal of tmp dir in test

Browse files
Files changed (1) hide show
  1. 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 test_multioutput_weighted_with_callable(self):
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(