diff --git a/olaflow/processing/animate.py b/olaflow/processing/animate.py index b06adbd..160e573 100644 --- a/olaflow/processing/animate.py +++ b/olaflow/processing/animate.py @@ -27,7 +27,9 @@ config.read(args.config) out = pathlib.Path(config.get("post", "out")) out.mkdir(parents=True, exist_ok=True) -with gzip.open(out.joinpath("pickle.gz"), "rb") as f: +with gzip.open(gzf, "rb") if ( + gzf := out.joinpath("pickle.gz") +).exists() else gzf.with_suffix("").open("rb") as f: model = pickle.load(f) x0 = config.getfloat("post", "x")