Avoid specifying np random state as type due to compatibility

MilesCranmer · Dec 24, 2023 · f2294b3 · f2294b3
1 parent a812e16
commit f2294b3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pysr/sr.py b/pysr/sr.py
@@ -724,7 +724,7 @@ def __init__(
         turbo: bool = False,
         precision: int = 32,
         enable_autodiff: bool = False,
-        random_state: Optional[Union[int, np.random.RandomState]] = None,
+        random_state=None,
         deterministic: bool = False,
         warm_start: bool = False,
         verbosity: int = 1,