Use importorskip in mlx tests

Author: jessegrabowski

.github/workflows/test.yml

@@ -289,7 +289,7 @@ jobs:
         shell: micromamba-shell {0}
         run: |
           export PYTENSOR_FLAGS=mode=FAST_COMPILE,warn__ignore_bug_before=all,on_opt_error=raise,on_shape_error=raise,gcc__cxxflags=-pipe
-          python -m pytest --runslow --benchmark-only --benchmark-json output.json --ignore tests/link/mlx
+          python -m pytest --runslow --benchmark-only --benchmark-json output.json
       - name: Store benchmark result
         uses: benchmark-action/github-action-benchmark@v1
         with:

tests/link/mlx/test_basic.py

@@ -5,7 +5,6 @@
 from collections.abc import Callable, Iterable
 from functools import partial
 
-import mlx.core as mx
 import numpy as np
 import pytest
 
@@ -16,13 +15,12 @@
 from pytensor.graph import RewriteDatabaseQuery
 from pytensor.graph.basic import Variable
 from pytensor.link.mlx import MLXLinker
-from pytensor.link.mlx.dispatch.core import (
-    mlx_funcify_Alloc,
-)
 from pytensor.raise_op import assert_op
 from pytensor.tensor.basic import Alloc
 
 
+mx = pytest.importorskip("mlx.core")
+
 optimizer = RewriteDatabaseQuery(include=["mlx"], exclude=MLX._optimizer.exclude)
 mlx_mode = Mode(linker=MLXLinker(), optimizer=optimizer)
 mlx_mode_no_compile = Mode(linker=MLXLinker(use_compile=False), optimizer=optimizer)
@@ -196,6 +194,9 @@ def test_alloc_with_different_shape_types():
     This addresses the TypeError that occurred when shape parameters
     contained MLX arrays instead of Python integers.
     """
+    from pytensor.link.mlx.dispatch.core import (
+        mlx_funcify_Alloc,
+    )
 
     # Create a mock node (we don't need a real node for this test)
     class MockNode:

tests/link/mlx/test_elemwise.py

@@ -1,11 +1,13 @@
-import mlx.core as mx
 import numpy as np
 import pytest
 
 import pytensor.tensor as pt
 from tests.link.mlx.test_basic import compare_mlx_and_py
 
 
+mx = pytest.importorskip("mlx.core")
+
+
 @pytest.mark.parametrize("op", [pt.any, pt.all, pt.max, pt.min])
 def test_input(op) -> None:
     x = pt.vector("x")

tests/link/mlx/test_math.py

@@ -1,4 +1,3 @@
-import mlx.core as mx
 import numpy as np
 import pytest
 
@@ -8,6 +7,9 @@
 from tests.link.mlx.test_basic import compare_mlx_and_py
 
 
+mx = pytest.importorskip("mlx.core")
+
+
 def test_dot():
     x = pt.matrix("x")
     y = pt.matrix("y")