fix ruff error

akotlar · Mar 14, 2024 · 5bbf225 · 5bbf225
1 parent ef4c79a
commit 5bbf225
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/python/python/bystro/ancestry/tests/test_listener.py b/python/python/bystro/ancestry/tests/test_listener.py
@@ -9,7 +9,7 @@
     SubmittedJobMessage,
     AncestryJobCompleteMessage,
     AncestryResults,
-    AncestryModels
+    AncestryModels,
 )
 from bystro.ancestry.tests.test_inference import (
     ANCESTRY_MODEL,
@@ -34,7 +34,10 @@ def test_submit_fn():
 
 
 def test_handler_fn_happy_path(mocker, tmpdir):
-    mocker.patch("bystro.ancestry.listener._get_models_from_s3", return_value=AncestryModels(ANCESTRY_MODEL, ANCESTRY_MODEL))
+    mocker.patch(
+        "bystro.ancestry.listener._get_models_from_s3",
+        return_value=AncestryModels(ANCESTRY_MODEL, ANCESTRY_MODEL),
+    )
     dosage_path = "some_dosage.feather"
     f1 = tmpdir.join(dosage_path)