From 3a21f61772d79d2d84671373dd14e4afbdaa6a80 Mon Sep 17 00:00:00 2001
From: Awni Hannun <awni@apple.com>
Date: Wed, 11 Dec 2024 23:56:25 -0800
Subject: [PATCH] Fix build (#1693)

---
 python/src/ops.cpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/python/src/ops.cpp b/python/src/ops.cpp
index bd64bf687..abfbbbc7c 100644
--- 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,