From 4a5f3b21bb535ae2b5639cdece27b368f7f1d576 Mon Sep 17 00:00:00 2001
From: Sugato Ray <sugatoray@users.noreply.github.com>
Date: Wed, 31 Jan 2024 15:05:42 -0500
Subject: [PATCH] Add `py.typed` to support PEP-561 (type-hinting) for `mlx`
 (#588)

* Add `py.typed` to support PEP-561 (type-hinting)

This adds support for type-hinting information as laid in [PEP-561](https://peps.python.org/pep-0561/).

* add py.typed to MANIFEST.in
---
 MANIFEST.in         | 1 +
 python/mlx/py.typed | 1 +
 2 files changed, 2 insertions(+)
 create mode 100644 python/mlx/py.typed

diff --git a/MANIFEST.in b/MANIFEST.in
index 3d116a511..d81234106 100644
--- a/MANIFEST.in
+++ b/MANIFEST.in
@@ -1,3 +1,4 @@
 include CMakeLists.txt
 recursive-include mlx/ *
 include python/src/*
+python/mlx/py.typed # support type hinting as in PEP-561
diff --git a/python/mlx/py.typed b/python/mlx/py.typed
new file mode 100644
index 000000000..8b1378917
--- /dev/null
+++ b/python/mlx/py.typed
@@ -0,0 +1 @@
+