fix: Add error handling for repo cloning

All-Hands-AI · May 19, 2024 · c79d0f2 · c79d0f2
1 parent 650c813
commit c79d0f2
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/swebench/harness/context_manager.py b/swebench/harness/context_manager.py
@@ -321,8 +321,10 @@ def __enter__(self):
                 # Clone github per repo/version
                 repo_path = os.path.join(self.testbed, env_name)
                 if not os.path.exists(repo_path):
-                    clone_repo(repo, repo_path)
-                    self.log.write(f"Cloned {repo} to {repo_path}")
+                    if clone_repo(repo, repo_path):
+                        self.log.write(f"Cloned {repo} to {repo_path}")
+                    else:
+                        raise Exception(f"Failed to clone {repo} to {repo_path}")
                 else:
                     self.log.write(f"Repo for {repo_prefix} version {version} exists: {repo_path}; skipping")