Spaces:
Sleeping
Sleeping
Fix tests on Julia 1.7 due to module assignment
Browse files- pysr/test/test.py +2 -2
pysr/test/test.py
CHANGED
|
@@ -125,8 +125,8 @@ class TestPipeline(unittest.TestCase):
|
|
| 125 |
model.fit(self.X, y)
|
| 126 |
|
| 127 |
# We should have that the model state is now a Float64 hof:
|
| 128 |
-
|
| 129 |
-
self.assertTrue(jl.
|
| 130 |
|
| 131 |
# Test options stored:
|
| 132 |
self.assertEqual(model.julia_options_.turbo, False)
|
|
|
|
| 125 |
model.fit(self.X, y)
|
| 126 |
|
| 127 |
# We should have that the model state is now a Float64 hof:
|
| 128 |
+
test_state = model.raw_julia_state_
|
| 129 |
+
self.assertTrue(jl.typeof(test_state[1]).parameters[1] == jl.Float64)
|
| 130 |
|
| 131 |
# Test options stored:
|
| 132 |
self.assertEqual(model.julia_options_.turbo, False)
|