mlx-cluster 0.0.3__tar.gz → 0.0.4__tar.gz

{mlx_cluster-0.0.3/mlx_cluster.egg-info → mlx_cluster-0.0.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: mlx_cluster
-Version: 0.0.3
+Version: 0.0.4
 Summary: C++ and Metal extensions for MLX CTC Loss
 Author-email: Vinay Pandya <vinayharshadpandya27@gmail.com>
 Project-URL: Homepage, https://github.com/vinayhpandya/mlx_cluster
@@ -15,9 +15,11 @@ Description-Content-Type: text/markdown
 License-File: LICENSE
 Provides-Extra: dev
 Provides-Extra: test
+Requires-Dist: mlx_graphs==0.0.7; extra == "test"
+Requires-Dist: torch==2.2.0; extra == "test"
+Requires-Dist: mlx>=0.17.0; extra == "test"
 Requires-Dist: pytest==7.4.4; extra == "test"
-Requires-Dist: torch; extra == "test"
-Requires-Dist: mlx_graphs; extra == "test"
+Requires-Dist: scipy==1.12.0; extra == "test"
 
 # mlx_cluster
 

{mlx_cluster-0.0.3 → mlx_cluster-0.0.4/mlx_cluster.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: mlx_cluster
-Version: 0.0.3
+Version: 0.0.4
 Summary: C++ and Metal extensions for MLX CTC Loss
 Author-email: Vinay Pandya <vinayharshadpandya27@gmail.com>
 Project-URL: Homepage, https://github.com/vinayhpandya/mlx_cluster
@@ -15,9 +15,11 @@ Description-Content-Type: text/markdown
 License-File: LICENSE
 Provides-Extra: dev
 Provides-Extra: test
+Requires-Dist: mlx_graphs==0.0.7; extra == "test"
+Requires-Dist: torch==2.2.0; extra == "test"
+Requires-Dist: mlx>=0.17.0; extra == "test"
 Requires-Dist: pytest==7.4.4; extra == "test"
-Requires-Dist: torch; extra == "test"
-Requires-Dist: mlx_graphs; extra == "test"
+Requires-Dist: scipy==1.12.0; extra == "test"
 
 # mlx_cluster
 

mlx_cluster-0.0.4/mlx_cluster.egg-info/requires.txt

@@ -0,0 +1,9 @@
+
+[dev]
+
+[test]
+mlx_graphs==0.0.7
+torch==2.2.0
+mlx>=0.17.0
+pytest==7.4.4
+scipy==1.12.0

{mlx_cluster-0.0.3 → mlx_cluster-0.0.4}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "mlx_cluster"
-version = "0.0.3"
+version = "0.0.4"
 authors = [
   { name = "Vinay Pandya", email = "vinayharshadpandya27@gmail.com" },
 ]
@@ -18,9 +18,11 @@ classifiers = [
 [project.optional-dependencies]
 dev = []
 test = [
+  "mlx_graphs==0.0.7",
+  "torch==2.2.0",
+  "mlx>=0.17.0",
   "pytest==7.4.4",
-  "torch",
-  "mlx_graphs",
+  "scipy==1.12.0",
 ]
 [project.urls]
 Homepage = "https://github.com/vinayhpandya/mlx_cluster"
@@ -31,9 +33,9 @@ Issues = "https://github.com/vinayhpandya/mlx_cluster/Issues"
 requires = [
   "setuptools>=42",
   "cmake>=3.24",
-  "mlx>=0.17.0",
+  "mlx==0.18.0",
   "nanobind@git+https://github.com/wjakob/nanobind.git@2f04eac452a6d9142dedb957701bdb20125561e4",
 ]
 
 
-build-backend = "setuptools.build_meta"
+build-backend = "setuptools.build_meta"

{mlx_cluster-0.0.3 → mlx_cluster-0.0.4}/random_walks/RandomWalk.cpp

@@ -137,6 +137,7 @@ std::vector<std::vector<int>> RandomWalk::output_shapes(const std::vector<array>
 
 array random_walk(const array& rowptr, const array& col, const array& start, const array& rand, int walk_length, StreamOrDevice s)
 {   
+    std::cout<<"Inside random walk"<<std::endl;
     int nodes = start.size();
     auto primitive = std::make_shared<RandomWalk>(walk_length, to_stream(s));
     return array::make_arrays({{nodes,walk_length+1},{nodes, walk_length}},

{mlx_cluster-0.0.3 → mlx_cluster-0.0.4}/setup.py

@@ -4,20 +4,13 @@ from mlx import extension
 if __name__ == "__main__":
     setup(
         name="mlx_cluster",
-        version="0.0.3",
+        version="0.0.4",
         description="Sample C++ and Metal extensions for MLX primitives.",
         ext_modules=[extension.CMakeExtension("mlx_cluster._ext")],
         cmdclass={"build_ext": extension.CMakeBuild},
         packages=["mlx_cluster"],
         package_data={"mlx_cluster": ["*.so", "*.dylib", "*.metallib"]},
-        extras_require={
-            "dev": [],
-            "test": [
-                "mlx_graphs",
-                "torch",
-                "pytest",
-            ],
-        },
+        extras_require={"dev": []},
         zip_safe=False,
         python_requires=">=3.8",
     )

{mlx_cluster-0.0.3 → mlx_cluster-0.0.4}/tests/test_random_walk.py

@@ -3,11 +3,13 @@ import numpy as np
 import time
 
 # Torch dataset
+import torch
 from torch.utils.data import DataLoader
 
 loader = DataLoader(range(2708), batch_size=2000)
 start_indices = next(iter(loader))
 
+
 from mlx_graphs.datasets import PlanetoidDataset
 from mlx_graphs.utils.sorting import sort_edge_index
 from torch.utils.data import DataLoader
@@ -16,6 +18,7 @@ from mlx_cluster import random_walk
 cora_dataset = PlanetoidDataset(name="cora", base_dir="~")
 # For some reason int_64t and int_32t are not compatible
 edge_index = cora_dataset.graphs[0].edge_index.astype(mx.int64)
+
 # Convert edge index into a CSR matrix
 sorted_edge_index = sort_edge_index(edge_index=edge_index)
 row_mlx = sorted_edge_index[0][0]

{mlx_cluster-0.0.3 → mlx_cluster-0.0.4}/tests/test_rejection_sampling.py

@@ -2,10 +2,10 @@ import mlx.core as mx
 import numpy as np
 import time
 
-# Torch dataloader
+# Torch dataset
+import torch
 from torch.utils.data import DataLoader
 
-
 loader = DataLoader(range(2708), batch_size=2000)
 start_indices = next(iter(loader))
 # random_walks = torch.ops.torch_cluster.random_walk(
@@ -14,6 +14,7 @@ start_indices = next(iter(loader))
 
 from mlx_graphs.datasets import PlanetoidDataset
 from mlx_graphs.utils.sorting import sort_edge_index
+from torch.utils.data import DataLoader
 from mlx_cluster import rejection_sampling
 
 cora_dataset = PlanetoidDataset(name="cora", base_dir="~")
@@ -30,5 +31,5 @@ start_time = time.time()
 node_sequence = rejection_sampling(
     row_ptr_mlx, col_mlx, start_indices, 5, 1.0, 3.0, stream=mx.cpu
 )
-print("Time taken to complete random walks : ", time.time() - start_time)
+print("Time taken to complete 1000 random walks : ", time.time() - start_time)
 print(node_sequence)

mlx_cluster-0.0.3/mlx_cluster.egg-info/requires.txt

@@ -1,7 +0,0 @@
-
-[dev]
-
-[test]
-pytest==7.4.4
-torch
-mlx_graphs