MilesCranmer commited on
Commit
6b6d1bb
1 Parent(s): 8b29fef

Explicitly pass kwargs to equation return

Browse files
Files changed (1) hide show
  1. pysr/sr.py +2 -2
pysr/sr.py CHANGED
@@ -277,7 +277,7 @@ def pysr(X=None, y=None, weights=None,
277
  if delete_tempfiles:
278
  shutil.rmtree(kwargs['tmpdir'])
279
 
280
- return get_hof()
281
 
282
 
283
  def _make_auxiliary_julia_str(julia_auxiliary_filenames, **kwargs):
@@ -596,7 +596,7 @@ def run_feature_selection(X, y, select_k_features):
596
  max_features=select_k_features, prefit=True)
597
  return selector.get_support(indices=True)
598
 
599
- def get_hof(equation_file=None, n_features=None, variable_names=None, extra_sympy_mappings=None):
600
  """Get the equations from a hall of fame file. If no arguments
601
  entered, the ones used previously from a call to PySR will be used."""
602
 
 
277
  if delete_tempfiles:
278
  shutil.rmtree(kwargs['tmpdir'])
279
 
280
+ return get_hof(**kwargs)
281
 
282
 
283
  def _make_auxiliary_julia_str(julia_auxiliary_filenames, **kwargs):
 
596
  max_features=select_k_features, prefit=True)
597
  return selector.get_support(indices=True)
598
 
599
+ def get_hof(equation_file=None, n_features=None, variable_names=None, extra_sympy_mappings=None, **kwargs):
600
  """Get the equations from a hall of fame file. If no arguments
601
  entered, the ones used previously from a call to PySR will be used."""
602