precommit fixes

Author: laurafitzgerald

src/codeflare_sdk/common/utils/test_generate_cert.py

@@ -191,7 +191,7 @@ def test_force_regenerate(mocker):
         "codeflare_sdk.common.utils.generate_cert._get_tls_base_dir",
         return_value=Path(os.getcwd()),
     )
-    
+
     # Generate certificates first time
     generate_tls_cert("test-regen", "default")
     tls_dir = Path(os.getcwd()) / "test-regen-default"
@@ -230,7 +230,7 @@ def test_refresh_tls_cert(mocker):
     Test the refresh_tls_cert function
     """
     from codeflare_sdk.common.utils.generate_cert import refresh_tls_cert
-    
+
     mocker.patch("kubernetes.config.load_kube_config", return_value="ignore")
     mocker.patch(
         "codeflare_sdk.common.utils.generate_cert.get_secret_name",
@@ -244,7 +244,7 @@ def test_refresh_tls_cert(mocker):
         "codeflare_sdk.common.utils.generate_cert._get_tls_base_dir",
         return_value=Path(os.getcwd()),
     )
-    
+
     # Generate initial certificates
     generate_tls_cert("refresh-test", "default")
     tls_dir = Path(os.getcwd()) / "refresh-test-default"

src/codeflare_sdk/ray/cluster/build_ray_cluster.py

@@ -187,9 +187,9 @@ def build_ray_cluster(cluster: "codeflare_sdk.ray.cluster.Cluster"):
         gcs_ft_options = {"redisAddress": cluster.config.redis_address}
 
         if cluster.config.external_storage_namespace:
-            gcs_ft_options["externalStorageNamespace"] = (
-                cluster.config.external_storage_namespace
-            )
+            gcs_ft_options[
+                "externalStorageNamespace"
+            ] = cluster.config.external_storage_namespace
 
         if cluster.config.redis_password_secret:
             gcs_ft_options["redisPassword"] = {
@@ -441,18 +441,28 @@ def head_worker_extended_resources_from_cluster(
         resource_type = cluster.config.extended_resource_mapping[k]
         if resource_type in FORBIDDEN_CUSTOM_RESOURCE_TYPES:
             continue
-        head_worker_extended_resources[0][resource_type] = (
-            cluster.config.head_extended_resource_requests[k]
-            + head_worker_extended_resources[0].get(resource_type, 0)
+        head_worker_extended_resources[0][
+            resource_type
+        ] = cluster.config.head_extended_resource_requests[
+            k
+        ] + head_worker_extended_resources[
+            0
+        ].get(
+            resource_type, 0
         )
 
     for k in cluster.config.worker_extended_resource_requests.keys():
         resource_type = cluster.config.extended_resource_mapping[k]
         if resource_type in FORBIDDEN_CUSTOM_RESOURCE_TYPES:
             continue
-        head_worker_extended_resources[1][resource_type] = (
-            cluster.config.worker_extended_resource_requests[k]
-            + head_worker_extended_resources[1].get(resource_type, 0)
+        head_worker_extended_resources[1][
+            resource_type
+        ] = cluster.config.worker_extended_resource_requests[
+            k
+        ] + head_worker_extended_resources[
+            1
+        ].get(
+            resource_type, 0
         )
     return head_worker_extended_resources
 

src/codeflare_sdk/ray/rayjobs/rayjob.py

@@ -281,9 +281,9 @@ def _build_rayjob_cr(self) -> Dict[str, Any]:
         # Add submitterPodTemplate if we have files to mount
         if files:
             secret_name = f"{self.name}-files"
-            rayjob_cr["spec"]["submitterPodTemplate"] = (
-                self._build_submitter_pod_template(files, secret_name)
-            )
+            rayjob_cr["spec"][
+                "submitterPodTemplate"
+            ] = self._build_submitter_pod_template(files, secret_name)
 
         # Configure cluster: either use existing or create new
         if self._cluster_config is not None: