format

Author: olupton

.github/triage/jax_toolbox_triage/args.py

@@ -175,19 +175,33 @@ def parse_args(args=None):
     )
     args = parser.parse_args(args=args)
     assert args.container_runtime in {"docker", "pyxis"}, args.container_runtime
-    passing_commits_known = (args.passing_container is not None) or (args.passing_commits is not None)
-    failing_commits_known = (args.failing_container is not None) or (args.failing_commits is not None)
+    passing_commits_known = (args.passing_container is not None) or (
+        args.passing_commits is not None
+    )
+    failing_commits_known = (args.failing_container is not None) or (
+        args.failing_commits is not None
+    )
     sets_of_known_commits = passing_commits_known + failing_commits_known
     if sets_of_known_commits == 2:
         # If the container-level search is being skipped, because a valid combination
         # of --{passing,failing}-{commits,container} is passed, then no container-level
         # search options should be passed.
-        assert args.container is None and args.start_date is None and args.end_date is None, "No container-level search options should be passed if the passing/failing containers/commits have been passed explicitly."
-        assert args.passing_container is not None or args.failing_container is not None, ""
+        assert (
+            args.container is None and args.start_date is None and args.end_date is None
+        ), (
+            "No container-level search options should be passed if the passing/failing containers/commits have been passed explicitly."
+        )
+        assert (
+            args.passing_container is not None or args.failing_container is not None
+        ), ""
     elif sets_of_known_commits == 1:
-        raise Exception("If --passing-{commits OR container} is passed then --failing-{commits OR container} should be too")
+        raise Exception(
+            "If --passing-{commits OR container} is passed then --failing-{commits OR container} should be too"
+        )
     else:
         # None of --{passing,failing}-{commits,container} were passed, make sure the
         # compulsory arguments for the container-level search were passed
-        assert args.container is not None, "--container must be passed for the container-level search"
+        assert args.container is not None, (
+            "--container must be passed for the container-level search"
+        )
     return args

.github/triage/jax_toolbox_triage/pyxis.py

@@ -11,6 +11,7 @@
 # for .exists()
 _process_token = secrets.token_bytes()
 
+
 class PyxisContainer:
     def __init__(
         self,

.github/triage/tests/test_arg_parsing.py

@@ -9,8 +9,18 @@
     "failing-url",
 ]
 valid_container_and_commits = [
-    ["--passing-container", "passing-url", "--failing-commits", "jax:0123456789,xla:fedcba9876543210"],
-    ["--failing-container", "failing-url", "--passing-commits", "xla:fedcba9876543210,jax:0123456789"],
+    [
+        "--passing-container",
+        "passing-url",
+        "--failing-commits",
+        "jax:0123456789,xla:fedcba9876543210",
+    ],
+    [
+        "--failing-container",
+        "failing-url",
+        "--passing-commits",
+        "xla:fedcba9876543210,jax:0123456789",
+    ],
 ]
 valid_start_end_date_args = [
     ["--container", "jax"],
@@ -21,7 +31,10 @@
 
 
 @pytest.mark.parametrize(
-    "good_args", [valid_start_end_container] + valid_start_end_date_args + valid_container_and_commits
+    "good_args",
+    [valid_start_end_container]
+    + valid_start_end_date_args
+    + valid_container_and_commits,
 )
 def test_good_container_args(good_args):
     args = parse_args(good_args + test_command)
@@ -48,7 +61,12 @@ def test_bad_container_arg_combinations_across_groups(date_args):
         ["--passing-container", "passing-url"],
         ["--failing-container", "failing-url"],
         # Need at least one container
-        ["--passing-commits", "jax:0123456789,xla:fedcba9876543210", "--failing-commits", "xla:fedcba9876543210,jax:0123456789"],
+        [
+            "--passing-commits",
+            "jax:0123456789,xla:fedcba9876543210",
+            "--failing-commits",
+            "xla:fedcba9876543210,jax:0123456789",
+        ],
         # Cannot combine --passing-container with --passing-commits etc.
         ["--passing-container", "passing-url", "--passing-commits", "jax:123,xla:456"],
         ["--failing-container", "failing-url", "--failing-commits", "jax:123,xla:456"],
@@ -76,6 +94,7 @@ def test_unparsable_container_args(container_args):
     with pytest.raises(SystemExit):
         parse_args(container_args + test_command)
 
+
 def test_invalid_container_runtime():
     with pytest.raises(Exception):
-        parse_args(["--container-runtime=magic-beans"] + test_command)
+        parse_args(["--container-runtime=magic-beans"] + test_command)