MilesCranmer commited on
Commit
c2b20b6
1 Parent(s): 859581c

Enforce use of keyword arguments

Browse files
Files changed (1) hide show
  1. pysr/sr.py +1 -4
pysr/sr.py CHANGED
@@ -347,6 +347,7 @@ class PySRRegressor(BaseEstimator, RegressorMixin):
347
  def __init__(
348
  self,
349
  model_selection="best",
 
350
  weights=None,
351
  binary_operators=None,
352
  unary_operators=None,
@@ -410,7 +411,6 @@ class PySRRegressor(BaseEstimator, RegressorMixin):
410
  precision=32,
411
  multithreading=None,
412
  use_symbolic_utils=False,
413
- **kwargs,
414
  ):
415
  """Initialize settings for an equation search in PySR.
416
 
@@ -530,8 +530,6 @@ class PySRRegressor(BaseEstimator, RegressorMixin):
530
  :type precision: int
531
  :param use_symbolic_utils: Whether to use SymbolicUtils during simplification.
532
  :type use_symbolic_utils: bool
533
- :param **kwargs: Other options passed to SymbolicRegression.Options, for example, if you modify SymbolicRegression.jl to include additional arguments.
534
- :type **kwargs: dict
535
  :returns: Initialized model. Call `.fit(X, y)` to fit your data!
536
  :type: PySRRegressor
537
  """
@@ -661,7 +659,6 @@ class PySRRegressor(BaseEstimator, RegressorMixin):
661
  multithreading=multithreading,
662
  use_symbolic_utils=use_symbolic_utils,
663
  ),
664
- **kwargs,
665
  }
666
 
667
  # Stored equations:
 
347
  def __init__(
348
  self,
349
  model_selection="best",
350
+ *,
351
  weights=None,
352
  binary_operators=None,
353
  unary_operators=None,
 
411
  precision=32,
412
  multithreading=None,
413
  use_symbolic_utils=False,
 
414
  ):
415
  """Initialize settings for an equation search in PySR.
416
 
 
530
  :type precision: int
531
  :param use_symbolic_utils: Whether to use SymbolicUtils during simplification.
532
  :type use_symbolic_utils: bool
 
 
533
  :returns: Initialized model. Call `.fit(X, y)` to fit your data!
534
  :type: PySRRegressor
535
  """
 
659
  multithreading=multithreading,
660
  use_symbolic_utils=use_symbolic_utils,
661
  ),
 
662
  }
663
 
664
  # Stored equations: