fix: specify install package dir

yoyololicon · Jul 8, 2023 · fdf6a7c · fdf6a7c
1 parent 4107604
commit fdf6a7c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -13,12 +13,12 @@
     long_description=long_description,
     long_description_content_type="text/markdown",
     url="https://github.com/yoyololicon/pytorch-NMFs",
-    packages=setuptools.find_packages(),
+    packages=["torchnmf"],
     install_requires=['torch>=1.4',
                       'tqdm'],
     classifiers=[
         "Programming Language :: Python :: 3",
         "License :: OSI Approved :: MIT License",
         "Operating System :: OS Independent",
     ],
-)
+)