Spaces:
Running
Running
Commit
·
e5b4869
1
Parent(s):
ccf71e9
Call `refresh` in load function
Browse files- pysr/sr.py +1 -2
pysr/sr.py
CHANGED
@@ -2090,7 +2090,6 @@ def load(
|
|
2090 |
**pysr_kwargs,
|
2091 |
)
|
2092 |
|
2093 |
-
model.equation_file_ = equation_file
|
2094 |
model.nout_ = nout
|
2095 |
model.n_features_in_ = n_features_in
|
2096 |
|
@@ -2105,6 +2104,6 @@ def load(
|
|
2105 |
else:
|
2106 |
model.selection_mask_ = selection_mask
|
2107 |
|
2108 |
-
model.refresh()
|
2109 |
|
2110 |
return model
|
|
|
2090 |
**pysr_kwargs,
|
2091 |
)
|
2092 |
|
|
|
2093 |
model.nout_ = nout
|
2094 |
model.n_features_in_ = n_features_in
|
2095 |
|
|
|
2104 |
else:
|
2105 |
model.selection_mask_ = selection_mask
|
2106 |
|
2107 |
+
model.refresh(checkpoint_file=equation_file)
|
2108 |
|
2109 |
return model
|