diff --git a/setup.py b/setup.py
index 1d1a7a0..4be43c8 100644
--- a/setup.py
+++ b/setup.py
@@ -36,7 +36,7 @@
     license="MIT",
     install_requires=requirements,
     include_package_data=True,
-    packages=find_packages(include=["KANama", "KANama.*"]),
+    # packages=find_packages(include=["KANama", "KANama.*", "model", "train"]),
     python_requires=">=3.10",
     classifiers=[
         "Programming Language :: Python :: 3.10"