fix test bug

CederGroupHub · May 6, 2024 · 9b31dbb · 9b31dbb
1 parent 3d5834e
commit 9b31dbb
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/tests/fake_lab/tasks/infinite_task.py b/tests/fake_lab/tasks/infinite_task.py
@@ -11,5 +11,6 @@ def __init__(self, samples: list[ObjectId], *args, **kwargs):
         self.sample = samples[0]
 
     def run(self):
-        with self.lab_view.request_resources({DeviceThatNeverEnds: {"infinite_loop": 1}}) as (device_that_fails, _):
-            device_that_fails.run_infinite()
+        with self.lab_view.request_resources({DeviceThatNeverEnds: {"infinite_loop": 1}}) as (devices, _):
+            device_that_never_ends = devices[DeviceThatNeverEnds]
+            device_that_never_ends.run_infinite()