format code

This commit is contained in:
stellahsr 2024-03-26 14:50:53 +08:00
parent 91db2ef112
commit e88b0fdf16
3 changed files with 3 additions and 4 deletions

View file

@ -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"

View file

@ -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

View file

@ -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 = {}):