Absolute imports

Author: jessegrabowski

tests/link/mlx/test_elemwise.py

@@ -1,8 +1,9 @@
+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
+from tests.link.mlx.test_basic import compare_mlx_and_py
 
 
 @pytest.mark.parametrize("op", [pt.any, pt.all, pt.max, pt.min])

tests/link/mlx/test_subtensor.py

@@ -1,10 +1,10 @@
 import numpy as np
 import pytest
-from test_basic import compare_mlx_and_py
 
 import pytensor.tensor as pt
 from pytensor.tensor import subtensor as pt_subtensor
 from pytensor.tensor import tensor
+from tests.link.mlx.test_basic import compare_mlx_and_py
 
 
 mx = pytest.importorskip("mlx.core")