Added yield statement to not return set_device_arm until after all te…

…sts done
brainglobe · May 29, 2024 · a31ee29 · a31ee29
1 parent 13c64e4
commit a31ee29
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/conftest.py b/tests/conftest.py
@@ -26,3 +26,5 @@ def set_device_arm_macos_ci():
         keras.src.backend.common.global_state.set_global_attribute(
             "torch_device", "cpu"
         )
+
+        yield