We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 218d2ad commit 0b4fb85Copy full SHA for 0b4fb85
pytensor/link/mlx/dispatch/random.py
@@ -5,7 +5,10 @@
5
6
import pytensor.tensor.random.basic as ptr
7
from pytensor.link.mlx.dispatch.basic import mlx_funcify, mlx_typify
8
-from pytensor.link.mlx.dispatch.core import convert_dtype_to_mlx, mlx_to_list_shape
+from pytensor.link.mlx.dispatch.tensor_basic import (
9
+ convert_dtype_to_mlx,
10
+ mlx_to_list_shape,
11
+)
12
13
14
def numpy_generator_to_mlx_key(rng: Generator) -> mx.array:
0 commit comments