Commit 6aba153d authored by Boris Bonev's avatar Boris Bonev
Browse files

fix to setup.py for explicit loading of the distributed package

parent ef4f6518
...@@ -36,7 +36,7 @@ setup( ...@@ -36,7 +36,7 @@ setup(
version='0.5', version='0.5',
author='Boris Bonev', author='Boris Bonev',
author_email='bbonev@nvidia.com', author_email='bbonev@nvidia.com',
packages=['torch_harmonics',], packages=['torch_harmonics','torch_harmonics.distributed'],
scripts=[], scripts=[],
url='https://github.com/NVIDIA/torch-harmonics', url='https://github.com/NVIDIA/torch-harmonics',
license='LICENSE.md', license='LICENSE.md',
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment