fix(RHOAIENG-20531): adjusted formatting

Signed-off-by: Pat O'Connor <[email protected]>

Author: chipspeak

poetry.lock

@@ -59,4 +59,4 @@ markers = [
 ]
 addopts = "--timeout=900"
 testpaths = ["src/codeflare_sdk"]
-collect_ignore = ["src/codeflare_sdk/common/utils/unit_test_support.py"]
+collect_ignore = ["src/codeflare_sdk/common/utils/unit_test_support.py"]

pyproject.toml

@@ -101,8 +101,7 @@ def test_aw_creation_local_queue(mocker):
         print("generated yaml")
         print(yaml.dump(aw_kueue, default_flow_style=False))
         print("expected yaml")
-        print(yaml.dump(expected_rc, default_flow_style=False)
-              )
+        print(yaml.dump(expected_rc, default_flow_style=False))
         assert aw_kueue == expected_rc
 
     # With resources loaded in memory, no Local Queue specified.

src/codeflare_sdk/common/kueue/test_kueue.py

@@ -40,7 +40,11 @@ def create_cluster_config(num_workers=2, write_to_file=False):
         worker_memory_limits=6,
         appwrapper=True,
         write_to_file=write_to_file,
-        annotations={"app.kubernetes.io/managed-by": "test-prefix", "key1": "value1", "key2": "value2"},
+        annotations={
+            "app.kubernetes.io/managed-by": "test-prefix",
+            "key1": "value1",
+            "key2": "value2",
+        },
     )
     return config
 
@@ -499,7 +503,11 @@ def create_cluster_all_config_params(mocker, cluster_name, is_appwrapper) -> Clu
         extended_resource_mapping={"example.com/gpu": "GPU", "intel.com/gpu": "TPU"},
         overwrite_default_resource_mapping=True,
         local_queue="local-queue-default",
-        annotations={"app.kubernetes.io/managed-by": "test-prefix", "key1": "value1", "key2": "value2"},
+        annotations={
+            "app.kubernetes.io/managed-by": "test-prefix",
+            "key1": "value1",
+            "key2": "value2",
+        },
         volumes=volumes,
         volume_mounts=volume_mounts,
     )

src/codeflare_sdk/common/utils/unit_test_support.py

@@ -145,7 +145,7 @@ def build_ray_cluster(cluster: "codeflare_sdk.ray.cluster.Cluster"):
                         cluster,
                         [get_head_container_spec(cluster)],
                         cluster.config.head_tolerations,
-                    )
+                    ),
                 },
             },
             "workerGroupSpecs": [
@@ -165,7 +165,7 @@ def build_ray_cluster(cluster: "codeflare_sdk.ray.cluster.Cluster"):
                             cluster,
                             [get_worker_container_spec(cluster)],
                             cluster.config.worker_tolerations,
-                        )
+                        ),
                     ),
                 }
             ],

src/codeflare_sdk/ray/cluster/build_ray_cluster.py

@@ -35,7 +35,17 @@ def test_default_cluster_creation(mocker):
     mocker.patch("kubernetes.client.ApisApi.get_api_versions")
     mocker.patch("kubernetes.client.CustomObjectsApi.list_namespaced_custom_object")
 
-    cluster = Cluster(ClusterConfiguration(name="default-cluster", namespace="ns", annotations={"app.kubernetes.io/managed-by": "test-prefix", "key1": "value1", "key2": "value2"}))
+    cluster = Cluster(
+        ClusterConfiguration(
+            name="default-cluster",
+            namespace="ns",
+            annotations={
+                "app.kubernetes.io/managed-by": "test-prefix",
+                "key1": "value1",
+                "key2": "value2",
+            },
+        )
+    )
 
     expected_rc = apply_template(
         f"{expected_clusters_dir}/ray/default-ray-cluster.yaml",
@@ -50,7 +60,16 @@ def test_default_appwrapper_creation(mocker):
     mocker.patch("kubernetes.client.CustomObjectsApi.list_namespaced_custom_object")
 
     cluster = Cluster(
-        ClusterConfiguration(name="default-appwrapper", namespace="ns", appwrapper=True, annotations={"app.kubernetes.io/managed-by": "test-prefix", "key1": "value1", "key2": "value2"})
+        ClusterConfiguration(
+            name="default-appwrapper",
+            namespace="ns",
+            appwrapper=True,
+            annotations={
+                "app.kubernetes.io/managed-by": "test-prefix",
+                "key1": "value1",
+                "key2": "value2",
+            },
+        )
     )
 
     expected_aw = apply_template(