From 45f6092dec6bb1e92fb53ff1ca36944fe9bcf618 Mon Sep 17 00:00:00 2001
From: Nicola De Angeli <112023843+niqodea@users.noreply.github.com>
Date: Wed, 1 Nov 2023 14:43:26 +0100
Subject: [PATCH] chore: add py.typed in compliance with PEP 561

---
 setup.py                  | 2 +-
 src/transformers/py.typed | 0
 2 files changed, 1 insertion(+), 1 deletion(-)
 create mode 100644 src/transformers/py.typed

diff --git a/setup.py b/setup.py
index 22407701f3c1..88737cc7d1e4 100644
--- a/setup.py
+++ b/setup.py
@@ -440,7 +440,7 @@ def run(self):
     package_dir={"": "src"},
     packages=find_packages("src"),
     include_package_data=True,
-    package_data={"": ["**/*.cu", "**/*.cpp", "**/*.cuh", "**/*.h", "**/*.pyx"]},
+    package_data={"": ["**/*.cu", "**/*.cpp", "**/*.cuh", "**/*.h", "**/*.pyx", "py.typed"]},
     zip_safe=False,
     extras_require=extras,
     entry_points={"console_scripts": ["transformers-cli=transformers.commands.transformers_cli:main"]},
diff --git a/src/transformers/py.typed b/src/transformers/py.typed
new file mode 100644
index 000000000000..e69de29bb2d1