From 7b31e101dab3ed469a6f135089f221f4bf2dfbce Mon Sep 17 00:00:00 2001
From: shubhra <shubhras01@gmail.com>
Date: Thu, 30 May 2024 14:08:52 +0530
Subject: [PATCH] adding in setup.py

---
 Pipfile                 | 9 +++++----
 examples/swe/try-swe.py | 1 -
 setup.py                | 6 +++++-
 3 files changed, 10 insertions(+), 6 deletions(-)

diff --git a/Pipfile b/Pipfile
index 4bfafdbf05..2403ad8063 100644
--- a/Pipfile
+++ b/Pipfile
@@ -11,12 +11,13 @@ termcolor = ">=2.4.0,<3"
 pydantic = ">=2.6.4,<3"
 openai = ">=1.3.0"
 rich = ">=13.7.1,<14"
-importlib-metadata = ">=4.8.1,<5"
+importlib-metadata = ">=4.8.1"
 cloup = ">=3.0.0"
 pyperclip = ">=1.8.2,<2"
-simple-parsing = "*"
-docker = "*"
-gymnasium = "*"
+simple-parsing = ">=0.1.5"
+docker = ">=7.1.0"
+gymnasium = ">=0.29.1"
+pyyaml = ">=6.0.1"
 
 [dev-packages]
 black = ">=24.1.1"
diff --git a/examples/swe/try-swe.py b/examples/swe/try-swe.py
index d4b435f68c..75b1d3759c 100644
--- a/examples/swe/try-swe.py
+++ b/examples/swe/try-swe.py
@@ -19,7 +19,6 @@
 
 if __name__ == "__main__":
     assert os.environ.get("GITHUB_ACCESS_TOKEN") is not None
-    assert os.environ.get("HARD_CODED_REPO_NAME") is not None
     # load config from YAML file
     task_config_path = script_dir / Path(CONFIG_FILE_PATH)
     with open(task_config_path, 'r') as stream:
diff --git a/setup.py b/setup.py
index 209e35dee8..4c88b9a8b7 100644
--- a/setup.py
+++ b/setup.py
@@ -38,10 +38,14 @@
         "pydantic>=2.6.4,<3",
         "openai>=1.3.0",
         "rich>=13.7.1,<14",
-        "importlib-metadata>=4.8.1,<5",
+        "importlib-metadata>=4.8.1",
         "pyperclip>=1.8.2,<2",
         "jsonref>=1.1.0",
         "inflection>=0.5.1",
+        "simple-parsing>=0.1.5",
+        "docker>=7.1.0",
+        "gymnasium>=0.29.1",
+        "pyyaml>=6.0.1",
     ],
     include_package_data=True,
 )