diff --git a/setup.py b/setup.py index 195fdb2e7be..880c36d9629 100644 --- a/setup.py +++ b/setup.py @@ -490,7 +490,7 @@ def run(self): # Package info packages=find_packages(exclude=('test',)), package_data={ - package_name: ['*.dll', '*.dylib', '*.so'] + package_name: ['*.dll', '*.dylib', '*.so', 'py.typed'], }, zip_safe=False, install_requires=requirements, diff --git a/torchvision/py.typed b/torchvision/py.typed new file mode 100644 index 00000000000..8b137891791 --- /dev/null +++ b/torchvision/py.typed @@ -0,0 +1 @@ +