fix: change to original directory after running verify install

aisingapore · Aug 16, 2022 · 18a4e2f · 18a4e2f
1 parent 78cd272
commit 18a4e2f
Showing 1 changed file with 3 additions and 0 deletions.
diff --git 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)