1 files changed, 12 insertions, 0 deletions
diff --git a/lib/utils.py b/lib/utils.py
index 86829b3..10f0172 100644
--- a/lib/utils.py
+++ b/lib/utils.py
@@ -2,13 +2,25 @@
 
 import json
 import numpy as np
+import os
 import re
 import logging
+from contextlib import contextmanager
 from sklearn.metrics import r2_score
 
 logger = logging.getLogger(__name__)
 
 
+@contextmanager
+def cd(path):
+    old_dir = os.getcwd()
+    os.chdir(path)
+    try:
+        yield
+    finally:
+        os.chdir(old_dir)
+
+
 class NpEncoder(json.JSONEncoder):
     def default(self, obj):
         if isinstance(obj, np.integer):