From a31ee292cd7d9db2bceb1d72b0a82c6095003583 Mon Sep 17 00:00:00 2001
From: Igor Tatarnikov <igor.tatarnikov@gmail.com>
Date: Wed, 29 May 2024 10:30:12 +0100
Subject: [PATCH] Added yield statement to not return set_device_arm until
 after all tests done

---
 tests/conftest.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/tests/conftest.py b/tests/conftest.py
index 50dffed..5dc1148 100644
--- 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