Spaces:
Sleeping
Sleeping
Fix `jl.test_state` issue
Browse files- pysr/test/test.py +5 -2
pysr/test/test.py
CHANGED
|
@@ -249,8 +249,11 @@ class TestPipeline(unittest.TestCase):
|
|
| 249 |
)
|
| 250 |
|
| 251 |
# We should have that the model state is now a Float32 hof:
|
| 252 |
-
|
| 253 |
-
self.assertTrue(
|
|
|
|
|
|
|
|
|
|
| 254 |
# This should exit almost immediately, and use the old equations
|
| 255 |
regressor.fit(X, y)
|
| 256 |
|
|
|
|
| 249 |
)
|
| 250 |
|
| 251 |
# We should have that the model state is now a Float32 hof:
|
| 252 |
+
test_state = regressor.julia_state_
|
| 253 |
+
self.assertTrue(
|
| 254 |
+
jl.first(jl.typeof(jl.last(test_state)).parameters) == jl.Float32
|
| 255 |
+
)
|
| 256 |
+
|
| 257 |
# This should exit almost immediately, and use the old equations
|
| 258 |
regressor.fit(X, y)
|
| 259 |
|