Additional logging messages during load

MilesCranmer · Jul 25, 2022 · c6902b7 · c6902b7
1 parent f1ac704
commit c6902b7
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/pysr/sr.py b/pysr/sr.py
@@ -2111,6 +2111,7 @@ def load(
     # Try to load model from <equation_file>.pkl
     print(f"Checking if {equation_file}.pkl exists...")
     if os.path.exists(str(equation_file) + ".pkl"):
+        print(f"Loading model from {equation_file}.pkl.")
         assert binary_operators is None
         assert unary_operators is None
         assert n_features_in is None