precommit

AD-SDL · Aug 7, 2024 · 1b87a2e · 1b87a2e
1 parent afcc131
commit 1b87a2e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_module.py b/tests/test_module.py
@@ -8,7 +8,7 @@
 import requests
 from wei import ExperimentClient
 from wei.types.module_types import ModuleAbout
-from wei.types.workcell_types import WorkcellData
+from wei.types.workcell_types import Workcell
 from wei.types.workflow_types import WorkflowStatus
 
 
@@ -22,7 +22,7 @@ def __init__(self, *args, **kwargs):
         self.workcell_file = self.root_dir / Path(
             "tests/workcell_defs/test_workcell.yaml"
         )
-        self.workcell = WorkcellData.from_yaml(self.workcell_file)
+        self.workcell = Workcell.from_yaml(self.workcell_file)
         self.server_host = self.workcell.config.server_host
         self.server_port = self.workcell.config.server_port
         self.url = f"http://{self.server_host}:{self.server_port}"