diff --git a/setup.py b/setup.py index 3222795a8..7b0ff08c5 100644 --- 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 new file mode 100644 index 000000000..e69de29bb