From d6a19d976b699e0b230b2e6c8fdd11a0c832ae83 Mon Sep 17 00:00:00 2001
From: Daniel Friesel <daniel.friesel@uos.de>
Date: Tue, 26 Oct 2021 13:18:59 +0200
Subject: kconfig loader: load kconfig from its base directory

this ensures that source statements are handled correctly
---
 lib/utils.py | 12 ++++++++++++
 1 file changed, 12 insertions(+)

(limited to 'lib/utils.py')

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):
-- 
cgit v1.2.3