fixes #13

unitaryai · Jan 22, 2021 · d869deb · d869deb
1 parent e994b57
commit d869deb
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -7,7 +7,7 @@
 
 setup(
     name="detoxify",
-    version="0.2.1",
+    version="0.2.2",
     description="A python library for detecting toxic comments",
     long_description=long_description,
     long_description_content_type="text/markdown",
@@ -19,7 +19,7 @@
         "torch >= 1.7.0",
         "sentencepiece >= 0.1.94"
     ],
-    packages=find_packages('detoxify', exclude=("tests", "src")),
+    packages=find_packages(include=["detoxify"], exclude=["tests", "src"]),
     classifiers=[
         "Programming Language :: Python :: 3",
         "License :: OSI Approved :: Apache Software License",