Fixes for setup files

cassidylaidlaw · Feb 21, 2024 · 5e08e62 · 5e08e62
1 parent a607a43
commit 5e08e62
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 8 deletions.
diff --git a/setup.cfg b/setup.cfg
@@ -1,3 +1,3 @@
 [metadata]
-description-file = README.md
-license_files=LICENSE.txt
+description_file = README.md
+license_files=LICENSE.txt
diff --git a/setup.py b/setup.py
@@ -1,5 +1,12 @@
 from distutils.core import setup
 
+with open("requirements.txt") as f:
+    requirements = []
+    for line in f:
+        # Exclude editable installs or requirements files
+        if not line.startswith(("-e", "-r", "git+", "http://", "https://")):
+            requirements.append(line.strip())
+
 setup(
     name="orpo",
     packages=[
@@ -25,12 +32,7 @@
         "AI safety",
         "reward misspecification",
     ],
-    install_requires=[
-        "torch>=1.13",
-        "numpy>=1.22",
-        "sacred>=0.8",
-        "ray[rllib]==2.7.1",
-    ],
+    install_requires=requirements,
     classifiers=[
         "Development Status :: 3 - Alpha",
         "Intended Audience :: Science/Research",