Add type hint stub (#780)

Summary: Fixes #779 ### Changes - Add `py.typed` file stub - Include the file into package data Pull Request resolved: #780 Reviewed By: NivekT Differential Revision: D39620230 Pulled By: ejguan fbshipit-source-id: 0661c08bfdaad66a5e50506937d1339fb2865fec
pytorch · Sep 19, 2022 · 2212cb7 · 2212cb7
1 parent 0b973a8
commit 2212cb7
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 0 deletions.
diff --git a/setup.py b/setup.py
@@ -182,6 +182,7 @@ def remove_extension(pattern):
         ],
         package_data={
             "torchdata": [
+                "py.typed",
                 "datapipes/iter/*.pyi",
                 "datapipes/map/*.pyi",
             ],

diff --git a/torchdata/py.typed b/torchdata/py.typed