fix: mlx import error is fixed

synnada-ai · Nov 19, 2024 · 2dca433 · 2dca433
1 parent aaf7dfe
commit 2dca433
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/tests/scripts/test_backend_fns.py b/tests/scripts/test_backend_fns.py
@@ -12,6 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+import platform
 from collections.abc import Callable
 from itertools import product
 
@@ -59,6 +60,8 @@
 try:
     import mlx.core as mx
 
+    if platform.system() != "Darwin":
+        raise ImportError
     testing_fns[MlxBackend] = mx.allclose
     installed_backends.append(MlxBackend)
 except ImportError: