diff --git a/setup.py b/setup.py index beebd807c7107..fcaa8ee97d706 100755 --- a/setup.py +++ b/setup.py @@ -76,7 +76,7 @@ def _load_py_module(fname, pkg="pytorch_lightning"): url=about.__homepage__, download_url='https://github.com/PyTorchLightning/pytorch-lightning', license=about.__license__, - packages=find_packages(exclude=['tests', 'tests/*', 'benchmarks', 'legacy', 'legacy/*']), + packages=find_packages(exclude=['tests', 'tests.*', 'benchmarks', 'legacy', 'legacy.*']), long_description=long_description, long_description_content_type='text/markdown', include_package_data=True,