Format for black

Author: milliams

citc/google.py

@@ -106,7 +106,8 @@ def get_types_info(client, nodespace):
             "memory": int(math.pow(mt["memoryMb"], 0.7) * 0.9 + 500),
             "cores_per_socket": mt["guestCpus"],
             "threads_per_core": "1",
-            "cluster_group": mt["name"].startswith("c2-")  # only C2 currently support compact placement groups
+            # only C2 currently support compact placement groups
+            "cluster_group": mt["name"].startswith("c2-"),
         }
         for mt in machine_types
     }

tests/test_list_nodes.py

@@ -11,7 +11,9 @@ def test_create_table():
         SlurmNode(name="foo-1", state="idle", state_flag=None, features={}, reason=""),
         SlurmNode(name="foo-2", state="idle", state_flag="~", features={}, reason=""),
     ]
-    cloud_nodes = [AwsNode(name="foo-1", state=NodeState.RUNNING, ip="10.0.0.25", id="i-foobar")]
+    cloud_nodes = [
+        AwsNode(name="foo-1", state=NodeState.RUNNING, ip="10.0.0.25", id="i-foobar")
+    ]
     table, headers = create_table(slurm_nodes, cloud_nodes)
     assert table == [
         ["foo-1", "idle", "", NodeState.RUNNING],
@@ -24,6 +26,8 @@ def test_print_table():
     slurm_nodes = [
         SlurmNode(name="foo-1", state="idle", state_flag=None, features={}, reason="")
     ]
-    cloud_nodes = [AwsNode(name="foo-1", state=NodeState.RUNNING, ip="10.0.0.25", id="i-foobar")]
+    cloud_nodes = [
+        AwsNode(name="foo-1", state=NodeState.RUNNING, ip="10.0.0.25", id="i-foobar")
+    ]
     table_data = create_table(slurm_nodes, cloud_nodes)
     print(tabulate(table_data))

tests/test_watchdog.py

@@ -17,7 +17,9 @@ def test_crosscheck_one_match():
     slurm_nodes = [
         SlurmNode(name="foo-1", state="idle", state_flag=None, features={}, reason="")
     ]
-    cloud_nodes = [AwsNode(name="foo-1", state=NodeState.RUNNING, ip="10.0.0.25", id="i-foobar")]
+    cloud_nodes = [
+        AwsNode(name="foo-1", state=NodeState.RUNNING, ip="10.0.0.25", id="i-foobar")
+    ]
 
     res = crosscheck(slurm_nodes, cloud_nodes)
     res = list(res)