specify package in setup file #13

unitaryai · Jan 22, 2021 · e994b57 · e994b57
1 parent 1c6d297
commit e994b57
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -19,7 +19,7 @@
         "torch >= 1.7.0",
         "sentencepiece >= 0.1.94"
     ],
-    packages=find_packages(exclude=("tests", "src")),
+    packages=find_packages('detoxify', exclude=("tests", "src")),
     classifiers=[
         "Programming Language :: Python :: 3",
         "License :: OSI Approved :: Apache Software License",