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

Fix pip install
codelion · Jan 29, 2025 · 5f507d0 · 5f507d0
2 parents 89eef8c + 64a5c81
commit 5f507d0
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/optillm/cepo/__init__.py b/optillm/cepo/__init__.py
diff --git a/setup.py b/setup.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=[