diff --git a/src/coffee/run.py b/src/coffee/run.py index 85f54d8f..b6960d7f 100644 --- a/src/coffee/run.py +++ b/src/coffee/run.py @@ -137,12 +137,12 @@ def update_standards_to(file): "platform": platform.platform(), "system": f"{platform.system()} {platform.release()} {platform.version()}", "node": platform.node(), - "python": platform.python_version() + "python": platform.python_version(), }, }, "standards": { "reference_sut": {"name": reference_sut.display_name, "id": reference_sut.key}, - "3_star": {harm.__class__.name(): harm_scores[harm].value() for harm in harm_scores.keys()} + "3_star": {harm.__class__.name(): harm_scores[harm].value() for harm in harm_scores.keys()}, }, } with open(file, "w") as out: diff --git a/tests/test_run.py b/tests/test_run.py index b53dbf7f..d8a7648d 100644 --- a/tests/test_run.py +++ b/tests/test_run.py @@ -16,4 +16,4 @@ def test_update_standards(fake_run, tmp_path): with open(new_path) as f: j = json.load(f) assert j["standards"]["3_star"][bias_harm.name()] == 0.123456 - assert j["standards"]["reference_sut"]["id"] == 'gpt2' + assert j["standards"]["reference_sut"]["id"] == "gpt2"