MilesCranmer commited on
Commit
c6902b7
1 Parent(s): f1ac704

Additional logging messages during load

Browse files
Files changed (1) hide show
  1. pysr/sr.py +1 -0
pysr/sr.py CHANGED
@@ -2111,6 +2111,7 @@ def load(
2111
  # Try to load model from <equation_file>.pkl
2112
  print(f"Checking if {equation_file}.pkl exists...")
2113
  if os.path.exists(str(equation_file) + ".pkl"):
 
2114
  assert binary_operators is None
2115
  assert unary_operators is None
2116
  assert n_features_in is None
 
2111
  # Try to load model from <equation_file>.pkl
2112
  print(f"Checking if {equation_file}.pkl exists...")
2113
  if os.path.exists(str(equation_file) + ".pkl"):
2114
+ print(f"Loading model from {equation_file}.pkl.")
2115
  assert binary_operators is None
2116
  assert unary_operators is None
2117
  assert n_features_in is None