diff --git a/setup.py b/setup.py index 8a67bcb2c5..1300af5908 100644 --- a/setup.py +++ b/setup.py @@ -72,8 +72,12 @@ def parse_args() -> argparse.Namespace: keywords=["pytorch", "torch", "training", "tools", "utilities"], python_requires=">=3.7", install_requires=read_requirements("requirements.txt"), + include_package_data=True, packages=find_packages(), package_data={"torchtnt": ["py.typed"]}, + data_files=[ + ('requirement_files',['requirements.txt', 'dev-requirements.txt']), + ], zip_safe=True, classifiers=[ "Development Status :: 2 - Pre-Alpha",