small inference bug fix

scripts/consolidate.py

@@ -186,8 +186,8 @@ def main():
     elif sweep_instances:
         raise NotImplementedError
 
+    plan = load_pickle(target_dir / "plan.pkl")
     if consolidate != 'copy':
-        plan = load_pickle(target_dir / "plan.pkl")
         plan["inference_plan"] = inference_plan
         save_pickle(plan, target_dir / "plan_inference.pkl")
 
@@ -203,7 +203,7 @@ def main():
                 )
     else:
         logger.warning("Plan used from fold 0, not updated with consolidation")
-
+        save_pickle(plan, target_dir / "plan_inference.pkl")
 
 if __name__ == '__main__':
     main()

scripts/predict.py

@@ -156,7 +156,8 @@ def main():
     cfg = OmegaConf.load(str(training_dir / "config.yaml"))
 
     cfg = set_arg(cfg, "task", task_name, force_args=force_args)
-    cfg["exp"] = set_arg(cfg["exp"], "fold", fold, force_args=force_args)
+    cfg["exp"] = set_arg(cfg["exp"], "fold", fold,
+                         force_args=True if fold == -1 else force_args)
     cfg["exp"] = set_arg(cfg["exp"], "id", model, force_args=force_args)
 
     overwrites = ov if ov is not None else []