diff --git a/swe_bench/inference/const.py b/swe_bench/inference/const.py index 84c5b0726..9611a209c 100644 --- a/swe_bench/inference/const.py +++ b/swe_bench/inference/const.py @@ -3,7 +3,7 @@ # @Desc : import pandas as pd -from metagpt.const import DATA_PATH, METAGPT_ROOT +from metagpt.const import METAGPT_ROOT SUBSET_DATASET = METAGPT_ROOT / "sub_swebench_dataset" / "sub_swebench.csv" SUBSET_DATASET_SKLERARN = METAGPT_ROOT / "sub_swebench_dataset" / "scikit-learn-68.csv" diff --git a/swe_bench/inference/run_api.py b/swe_bench/inference/run_api.py index 2ffe87c5b..37d87d6ce 100644 --- a/swe_bench/inference/run_api.py +++ b/swe_bench/inference/run_api.py @@ -2,15 +2,14 @@ import json from pathlib import Path import fire - from tqdm.auto import tqdm from metagpt.config2 import config from metagpt.logs import logger from metagpt.utils import count_string_tokens +from swe_bench.data.load_dataset import load_oracle_dataset from swe_bench.inference.run_agent import run_instance from swe_bench.utils.utils import check_existing_ids, extract_diff -from swe_bench.data.load_dataset import load_oracle_dataset # Replace with your own MAX_TOKEN = 128000 diff --git a/swe_bench/make_datasets/make_dataset.py b/swe_bench/make_datasets/make_dataset.py index c5df53b1d..233f6b01c 100644 --- a/swe_bench/make_datasets/make_dataset.py +++ b/swe_bench/make_datasets/make_dataset.py @@ -7,10 +7,10 @@ from pathlib import Path from tqdm.auto import tqdm from metagpt.logs import logger +from swe_bench.inference.const import TESTBED from swe_bench.make_datasets.make_instance import prompt_style_2_edits_only from swe_bench.utils.parse_diff import filter_changed_line from swe_bench.utils.repo_utils import EnvManager -from swe_bench.inference.const import TESTBED def reset_task_env(instance: dict = {}):