diff --git a/tests/test_all_pipeline.py b/tests/test_all_pipeline.py index f4fdc50ff..21a82cd30 100644 --- a/tests/test_all_pipeline.py +++ b/tests/test_all_pipeline.py @@ -163,6 +163,7 @@ def train_with_sigana(): sar.generate() ic = sar.load(sar.get_path("ic.pkl")) ric = sar.load(sar.get_path("ric.pkl")) + pred_score = sar.load("pred.pkl") uri_path = R.get_uri() return pred_score, {"ic": ic, "ric": ric}, uri_path