Skip to content

Commit 4fa4827

Browse files
set allow-pickle=true to load saved score arrays.
1 parent eb0f28f commit 4fa4827

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

src/sail/pipeline/pipeline.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -528,7 +528,7 @@ def load(cls, model_folder, name="sail_pipeline"):
528528
# Pre-load progressive scorer state from initial_points if exist.
529529
# -------------------------------------------
530530
if os.path.exists(os.path.join(load_location, "scorer_state.npz")):
531-
initial_points = np.load(os.path.join(load_location, "scorer_state.npz"))
531+
initial_points = np.load(os.path.join(load_location, "scorer_state.npz"), allow_pickle=True)
532532
sail_pipeline._scorer.set_state(initial_points)
533533
# sail_pipeline._scorer.progressive_score(
534534
# initial_points["y_true"], initial_points["y_pred"], verbose=1

0 commit comments

Comments
 (0)