[fix] fix potential ModuleNotFoundError

tswsxk · tswsxk · commit dd368ebf2ea1 · 2021-05-24T18:09:01.000+08:00
diff --git a/CHANGE.txt b/CHANGE.txt
@@ -1,3 +1,6 @@
+v0.0.4:
+   * fix potential ModuleNotFoundError
+
 v0.0.3:
     * add DKT
     
diff --git a/setup.py b/setup.py
@@ -1,4 +1,4 @@
-from setuptools import setup
+from setuptools import setup, find_packages
 
 test_deps = [
     'pytest>=4',
@@ -8,10 +8,11 @@
 
 setup(
     name='EduKTM',
-    version='0.0.3',
+    version='0.0.4',
     extras_require={
         'test': test_deps,
     },
+    packages=find_packages(),
     install_requires=[
         "torch",
         "tqdm",
