Fix build (#1693)

ml-explore · Dec 12, 2024 · 3a21f61 · 3a21f61
1 parent 4e1e952
commit 3a21f61
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/src/ops.cpp b/python/src/ops.cpp
@@ -105,7 +105,7 @@ void init_ops(nb::module_& m) {
   )pbdoc");
   m.def(
       "unflatten",
-      &unflatten,
+      &mx::unflatten,
       nb::arg(),
       "axis"_a,
       "shape"_a,