From 18a4e2f308ce37e5ece45c8fdc097bbfe1d92d5f Mon Sep 17 00:00:00 2001
From: yier <56420072+liyier90@users.noreply.github.com>
Date: Tue, 16 Aug 2022 13:33:54 +0800
Subject: [PATCH] fix: change to original directory after running verify
 install

---
 peekingduck/cli.py | 3 +++
 1 file changed, 3 insertions(+)

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)