MilesCranmer commited on
Commit
f2294b3
1 Parent(s): a812e16

Avoid specifying np random state as type due to compatibility

Browse files
Files changed (1) hide show
  1. pysr/sr.py +1 -1
pysr/sr.py CHANGED
@@ -724,7 +724,7 @@ class PySRRegressor(MultiOutputMixin, RegressorMixin, BaseEstimator):
724
  turbo: bool = False,
725
  precision: int = 32,
726
  enable_autodiff: bool = False,
727
- random_state: Optional[Union[int, np.random.RandomState]] = None,
728
  deterministic: bool = False,
729
  warm_start: bool = False,
730
  verbosity: int = 1,
 
724
  turbo: bool = False,
725
  precision: int = 32,
726
  enable_autodiff: bool = False,
727
+ random_state=None,
728
  deterministic: bool = False,
729
  warm_start: bool = False,
730
  verbosity: int = 1,