Fix build (#1693)

2025-10-19 00:04:41 +08:00 · 2024-12-11 23:56:25 -08:00
parent 4e1e9520e1
commit 3a21f61772
1 changed files with 1 additions and 1 deletions
--- 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,