format

Signed-off-by: oliver könig <okoenig@nvidia.com>

Author: ko3n1g

test/run/ray/test_kuberay.py

@@ -2102,9 +2102,7 @@ def test_kuberay_cluster_fallback_to_incluster(self):
         with patch(
             "nemo_run.run.ray.kuberay.config.load_kube_config", side_effect=kube_error
         ) as mock_load_kube:
-            with patch(
-                "nemo_run.run.ray.kuberay.config.load_incluster_config"
-            ) as mock_incluster:
+            with patch("nemo_run.run.ray.kuberay.config.load_incluster_config") as mock_incluster:
                 with patch("nemo_run.run.ray.kuberay.client.CustomObjectsApi"):
                     with patch("nemo_run.run.ray.kuberay.client.CoreV1Api"):
                         with patch("nemo_run.run.ray.kuberay.get_user", return_value="testuser"):
@@ -2121,9 +2119,7 @@ def test_kuberay_cluster_both_configs_fail(self):
         kube_error = Exception("Kube config file not found")
         incluster_error = Exception("Not running inside a cluster")
 
-        with patch(
-            "nemo_run.run.ray.kuberay.config.load_kube_config", side_effect=kube_error
-        ):
+        with patch("nemo_run.run.ray.kuberay.config.load_kube_config", side_effect=kube_error):
             with patch(
                 "nemo_run.run.ray.kuberay.config.load_incluster_config",
                 side_effect=incluster_error,
@@ -2160,9 +2156,7 @@ def test_kuberay_job_fallback_to_incluster(self):
         with patch(
             "nemo_run.run.ray.kuberay.config.load_kube_config", side_effect=kube_error
         ) as mock_load_kube:
-            with patch(
-                "nemo_run.run.ray.kuberay.config.load_incluster_config"
-            ) as mock_incluster:
+            with patch("nemo_run.run.ray.kuberay.config.load_incluster_config") as mock_incluster:
                 with patch("nemo_run.run.ray.kuberay.client.CustomObjectsApi"):
                     with patch("nemo_run.run.ray.kuberay.client.CoreV1Api"):
                         with patch("nemo_run.run.ray.kuberay.get_user", return_value="testuser"):
@@ -2179,9 +2173,7 @@ def test_kuberay_job_both_configs_fail(self):
         kube_error = Exception("Kube config file not found")
         incluster_error = Exception("Not running inside a cluster")
 
-        with patch(
-            "nemo_run.run.ray.kuberay.config.load_kube_config", side_effect=kube_error
-        ):
+        with patch("nemo_run.run.ray.kuberay.config.load_kube_config", side_effect=kube_error):
             with patch(
                 "nemo_run.run.ray.kuberay.config.load_incluster_config",
                 side_effect=incluster_error,
@@ -2200,9 +2192,7 @@ def test_error_chaining_preserved(self):
         kube_error = Exception("Kube config file not found")
         incluster_error = Exception("Not running inside a cluster")
 
-        with patch(
-            "nemo_run.run.ray.kuberay.config.load_kube_config", side_effect=kube_error
-        ):
+        with patch("nemo_run.run.ray.kuberay.config.load_kube_config", side_effect=kube_error):
             with patch(
                 "nemo_run.run.ray.kuberay.config.load_incluster_config",
                 side_effect=incluster_error,