Add py.typed (#37022)

2025-04-02 21:17:27 +08:00
parent 32c12aaec3
commit 7613cf1a45
2 changed files with 1 additions and 1 deletions
--- a/setup.py
+++ b/setup.py
@@ -458,7 +458,7 @@ setup(
    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"]},