Skip to content

Commit 0b4fb85

Browse files
committed
fix import after core.py renamed to tensor_basic.py
1 parent 218d2ad commit 0b4fb85

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

pytensor/link/mlx/dispatch/random.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,10 @@
55

66
import pytensor.tensor.random.basic as ptr
77
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
8+
from pytensor.link.mlx.dispatch.tensor_basic import (
9+
convert_dtype_to_mlx,
10+
mlx_to_list_shape,
11+
)
912

1013

1114
def numpy_generator_to_mlx_key(rng: Generator) -> mx.array:

0 commit comments

Comments
 (0)