Просмотр исходного кода

pass swe-bench box config parameter (#3189)

Engel Nyst 1 год назад
Родитель
Сommit
93433fa849
2 измененных файлов с 2 добавлено и 1 удалено
  1. 1 0
      evaluation/swe_bench/run_infer.py
  2. 1 1
      evaluation/swe_bench/swe_env_box.py

+ 1 - 0
evaluation/swe_bench/run_infer.py

@@ -211,6 +211,7 @@ def process_instance(
     # You can omit this if you don't need to setup specialized sandbox
     workspace_dir_name = f'{instance.repo}__{instance.version}'.replace('/', '__')
     sandbox = SWEBenchSSHBox.get_box_for_instance(
+        config,
         instance,
         workspace_dir_name,
         workspace_mount_path=workspace_mount_path,

+ 1 - 1
evaluation/swe_bench/swe_env_box.py

@@ -146,8 +146,8 @@ class SWEBenchSSHBox(DockerSSHBox):
     @classmethod
     def get_box_for_instance(
         cls,
-        instance,
         config: AppConfig,
+        instance,
         workspace_dir_name=None,
         skip_workspace_mount: bool = True,
         workspace_mount_path: str | None = None,