Fix jl.test_state issue

MilesCranmer · Feb 12, 2024 · 7c2bce0 · 7c2bce0
1 parent 6e07d35
commit 7c2bce0
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/pysr/test/test.py b/pysr/test/test.py
@@ -249,8 +249,11 @@ def test_empty_operators_single_input_warm_start(self):
         )
 
         # We should have that the model state is now a Float32 hof:
-        jl.test_state = regressor.raw_julia_state_
-        self.assertTrue(jl.seval("typeof(test_state[2]).parameters[1] == Float32"))
+        test_state = regressor.julia_state_
+        self.assertTrue(
+            jl.first(jl.typeof(jl.last(test_state)).parameters) == jl.Float32
+        )
+
         # This should exit almost immediately, and use the old equations
         regressor.fit(X, y)