diff --git a/peekingduck/cli.py b/peekingduck/cli.py index a6029c55b..2e910e05b 100644 --- a/peekingduck/cli.py +++ b/peekingduck/cli.py @@ -18,6 +18,7 @@ import logging import math +import os import tempfile from pathlib import Path from time import perf_counter @@ -430,6 +431,7 @@ def _verify_install() -> None: """ LoggerSetup.set_log_level("info") + cwd = Path.cwd() with tempfile.TemporaryDirectory() as tmp_dir: pipeline_config_path = Path(tmp_dir) / "verification_pipeline.yml" @@ -456,3 +458,4 @@ def _verify_install() -> None: num_iter=None, ) runner.run() + os.chdir(cwd)