specify .nn.utils in setup

hjbiao09 · Feb 29, 2020 · bac06f8 · bac06f8
1 parent 8527a8f
commit bac06f8
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -10,10 +10,11 @@
     author_email="[email protected]",
     packages=[
         "cplxmodule",
-        "cplxmodule.utils",
         "cplxmodule.nn",
         "cplxmodule.nn.relevance",
-        "cplxmodule.nn.masked"
+        "cplxmodule.nn.masked",
+        "cplxmodule.nn.utils",
+        "cplxmodule.utils",
     ],
     requires=["torch", "numpy"]
 )