refactor: more typing info

MilesCranmer · Nov 30, 2024 · 324d1f6 · 324d1f6
1 parent 9e87e03
commit 324d1f6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pysr/sr.py b/pysr/sr.py
@@ -841,7 +841,7 @@ def __init__(
         bumper: bool = False,
         precision: Literal[16, 32, 64] = 32,
         enable_autodiff: bool = False,
-        random_state=None,
+        random_state: int | np.random.RandomState | None = None,
         deterministic: bool = False,
         warm_start: bool = False,
         verbosity: int = 1,