Avoid circular import error

ImperialCollegeLondon · Oct 2, 2024 · 2a92f04 · 2a92f04
1 parent 4f0214f
commit 2a92f04
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/tests/test_nodes.py b/tests/test_nodes.py
@@ -422,15 +422,15 @@ def test_data_read(self):
     def test_data_overrides(self):
         data_path = "../docs/demo/data/processed/example_override_data.csv.gz"
         input_data = pd.read_csv(data_path)
-        
-        overrides = {'data_input_dict': data_path}
+
+        overrides = {"data_input_dict": data_path}
         node = Node(name="")
         node.apply_overrides(overrides)
         node.t = list(node.data_input_dict.keys())[0][1]
 
         self.assertEqual(
             input_data.groupby("variable").get_group("temperature")["value"].iloc[0],
-            node.get_data_input("temperature")
+            node.get_data_input("temperature"),
         )
 
 

diff --git a/wsimod/nodes/nodes.py b/wsimod/nodes/nodes.py
@@ -10,7 +10,6 @@
 
 from wsimod.core import constants
 from wsimod.core.core import WSIObj
-from wsimod.orchestration.model import read_csv
 
 
 class Node(WSIObj):
@@ -89,7 +88,8 @@ def apply_overrides(self, overrides: Dict[str, Any] = {}) -> None:
             overrides (dict, optional): Dictionary of overrides. Defaults to {}.
         """
         # overrides data_input_dict
-
+        from wsimod.orchestration.model import read_csv
+
         content = overrides.pop("data_input_dict", self.data_input_dict)
         if isinstance(content, str):
             self.data_input_dict = read_csv(content)