fix: lint

Author: pawelpaszki

tests/e2e/heterogeneous_clusters_kind_test.py

@@ -33,8 +33,13 @@ def run_heterogeneous_clusters(
     ):
         # Use GPU-enabled Ray image when GPUs are requested
         from codeflare_sdk.common.utils import constants
-        ray_image = f"rayproject/ray:{constants.RAY_VERSION}-gpu" if number_of_gpus > 0 else f"rayproject/ray:{constants.RAY_VERSION}"
-        
+
+        ray_image = (
+            f"rayproject/ray:{constants.RAY_VERSION}-gpu"
+            if number_of_gpus > 0
+            else f"rayproject/ray:{constants.RAY_VERSION}"
+        )
+
         for flavor in self.resource_flavors:
             node_labels = (
                 get_flavor_spec(self, flavor).get("spec", {}).get("nodeLabels", {})

tests/e2e/local_interactive_sdk_kind_test.py

@@ -51,7 +51,12 @@ def run_local_interactives(
 
         # Use GPU-enabled Ray image when GPUs are requested
         from codeflare_sdk.common.utils import constants
-        ray_image = f"rayproject/ray:{constants.RAY_VERSION}-gpu" if number_of_gpus > 0 else f"rayproject/ray:{constants.RAY_VERSION}"
+
+        ray_image = (
+            f"rayproject/ray:{constants.RAY_VERSION}-gpu"
+            if number_of_gpus > 0
+            else f"rayproject/ray:{constants.RAY_VERSION}"
+        )
 
         cluster = Cluster(
             ClusterConfiguration(

tests/e2e/mnist_raycluster_sdk_aw_kind_test.py

@@ -39,8 +39,13 @@ def run_mnist_raycluster_sdk_kind(
     ):
         # Use GPU-enabled Ray image when GPUs are requested
         from codeflare_sdk.common.utils import constants
-        ray_image = f"rayproject/ray:{constants.RAY_VERSION}-gpu" if number_of_gpus > 0 else f"rayproject/ray:{constants.RAY_VERSION}"
-        
+
+        ray_image = (
+            f"rayproject/ray:{constants.RAY_VERSION}-gpu"
+            if number_of_gpus > 0
+            else f"rayproject/ray:{constants.RAY_VERSION}"
+        )
+
         cluster = Cluster(
             ClusterConfiguration(
                 name="mnist",

tests/e2e/mnist_raycluster_sdk_kind_test.py

@@ -39,8 +39,13 @@ def run_mnist_raycluster_sdk_kind(
     ):
         # Use GPU-enabled Ray image when GPUs are requested
         from codeflare_sdk.common.utils import constants
-        ray_image = f"rayproject/ray:{constants.RAY_VERSION}-gpu" if number_of_gpus > 0 else f"rayproject/ray:{constants.RAY_VERSION}"
-        
+
+        ray_image = (
+            f"rayproject/ray:{constants.RAY_VERSION}-gpu"
+            if number_of_gpus > 0
+            else f"rayproject/ray:{constants.RAY_VERSION}"
+        )
+
         cluster = Cluster(
             ClusterConfiguration(
                 name="mnist",