From d96a2bdf570a21ef0cc2089bf5772271aefd21e9 Mon Sep 17 00:00:00 2001
From: Cheng <zcbenz@gmail.com>
Date: Mon, 19 Jan 2026 09:34:02 +0900
Subject: [PATCH] Fix python package install path in stubgen (#3009)

---
 python/src/CMakeLists.txt | 1 +
 1 file changed, 1 insertion(+)

diff --git a/python/src/CMakeLists.txt b/python/src/CMakeLists.txt
index 1a70bf4a..69152f50 100644
--- a/python/src/CMakeLists.txt
+++ b/python/src/CMakeLists.txt
@@ -37,6 +37,7 @@ if(MLX_BUILD_PYTHON_STUBS)
     MODULE
     "mlx.core"
     PYTHON_PATH
+    "$<TARGET_FILE_DIR:core>/.."
     "${CMAKE_CURRENT_SOURCE_DIR}/.."
     PATTERN_FILE
     "${CMAKE_CURRENT_SOURCE_DIR}/../mlx/_stub_patterns.txt"