From 4e02d31cdccc3305f8b66a61e3e40189e5ebe4eb Mon Sep 17 00:00:00 2001
From: Matthias <xmatthias@outlook.com>
Date: Sun, 10 Mar 2024 18:20:54 +0100
Subject: [PATCH] Mock torch on mac ARM

---
 tests/freqai/conftest.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/freqai/conftest.py b/tests/freqai/conftest.py
index 55f0296a3..6f3c3018d 100644
--- a/tests/freqai/conftest.py
+++ b/tests/freqai/conftest.py
@@ -33,7 +33,7 @@ def is_arm() -> bool:
 @pytest.fixture(autouse=True)
 def patch_torch_initlogs(mocker) -> None:
 
-    if is_mac() and not is_arm():
+    if is_mac():
         # Mock torch import completely
         import sys
         import types