fix submodule stubs (#1492)

2024-10-15 16:23:37 -07:00
parent 3f86399922
commit f9f8c167d4
3 changed files with 24 additions and 2 deletions
@@ -0,0 +1,20 @@
+mlx.core.distributed.__prefix__:
+  from mlx.core import array, Dtype, Device, Stream
+  from mlx.core.distributed import Group
+  from typing import Sequence, Optional, Union
+
+mlx.core.fast.__prefix__:
+  from mlx.core import array, Dtype, Device, Stream
+  from typing import Sequence, Optional, Union
+
+mlx.core.linalg.__prefix__:
+  from mlx.core import array, Dtype, Device, Stream
+  from typing import Sequence, Optional, Tuple, Union
+
+mlx.core.metal.__prefix__:
+  from mlx.core import array, Dtype, Device, Stream
+  from typing import Sequence, Optional, Union
+
+mlx.core.random.__prefix__:
+  from mlx.core import array, Dtype, Device, Stream
+  from typing import Sequence, Optional, Union