More changes for black.

mlcommons · Feb 21, 2024 · 644a977 · 644a977
1 parent 5e9bd67
commit 644a977
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git 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
@@ -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"