Merge pull request #160 from codelion/fix-pip-install

Fix pip install

Author: codelion

optillm/cepo/__init__.py

@@ -2,12 +2,15 @@
 
 setup(
     name="optillm",
-    version="0.1.1",
+    version="0.1.2",
     packages=find_packages(),
     py_modules=['optillm'],
     package_data={
-        'optillm': ['plugins/*.py'],  # Include plugin files
-        'optillm': ['cepo/configs/*.yaml'],  # Include yaml files in the package
+        'optillm': [
+            'plugins/*.py',  # Include plugin files
+            'cepo/*.py',     # Include cepo module Python files
+            'cepo/configs/*.yaml',  # Include yaml files
+        ],
     },
     include_package_data=True,  # This is important
     install_requires=[