gmakstutis/cognee
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Fix merge conflicts

Browse source
This commit is contained in:
Leon Luithlen 2024-11-29 11:33:05 +01:00
parent b46af5a6f6
commit d9fc740ec0
1 changed files with 0 additions and 20 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

20
evals/eval_swe_bench.py
View file

@ -45,15 +45,7 @@ def check_install_package(package_name):
except subprocess.CalledProcessError: except subprocess.CalledProcessError:
return False return False
<<<<<<< HEAD
async def generate_patch_with_cognee(instance, search_type=SearchType.CHUNKS): async def generate_patch_with_cognee(instance, search_type=SearchType.CHUNKS):
=======
async def generate_patch_with_cognee(
instance, search_type=SearchType.CHUNKS
):
>>>>>>> c4e3634 (Update eval_swe_bench)
await cognee.prune.prune_data() await cognee.prune.prune_data()
await cognee.prune.prune_system() await cognee.prune.prune_system()
@ -137,10 +129,7 @@ async def get_preds(dataset, with_cognee=True):
for instance in dataset for instance in dataset
] ]
model_patches = await asyncio.gather(*[x[1] for x in futures]) model_patches = await asyncio.gather(*[x[1] for x in futures])
<<<<<<< HEAD
=======
>>>>>>> c4e3634 (Update eval_swe_bench)
preds = [ preds = [
{ {
"instance_id": instance_id, "instance_id": instance_id,
@ -185,7 +174,6 @@ async def main():
with open(predictions_path, "w") as file: with open(predictions_path, "w") as file:
json.dump(preds, file) json.dump(preds, file)
<<<<<<< HEAD
subprocess.run( subprocess.run(
[ [
@ -205,14 +193,6 @@ async def main():
] ]
) )
=======
subprocess.run(["python", "-m", "swebench.harness.run_evaluation",
"--dataset_name", dataset_name,
"--split", "test",
"--predictions_path", predictions_path,
"--max_workers", "1",
"--run_id", "test_run"])
>>>>>>> c4e3634 (Update eval_swe_bench)
if __name__ == "__main__": if __name__ == "__main__":
import asyncio import asyncio