diff --git a/gateway/api/ray.py b/gateway/api/ray.py index 9131c80c6..3fb8aa4ca 100644 --- a/gateway/api/ray.py +++ b/gateway/api/ray.py @@ -227,9 +227,7 @@ def create_ray_cluster( config.load_incluster_config() k8s_client = kubernetes_client.api_client.ApiClient() dyn_client = DynamicClient(k8s_client) - raycluster_client = dyn_client.resources.get( - api_version="v1", kind="RayCluster" - ) + raycluster_client = dyn_client.resources.get(api_version="v1", kind="RayCluster") response = raycluster_client.create(body=cluster_data, namespace=namespace) if response.metadata.name != cluster_name: raise RuntimeError( @@ -287,9 +285,7 @@ def kill_ray_cluster(cluster_name: str) -> bool: config.load_incluster_config() k8s_client = kubernetes_client.api_client.ApiClient() dyn_client = DynamicClient(k8s_client) - raycluster_client = dyn_client.resources.get( - api_version="v1", kind="RayCluster" - ) + raycluster_client = dyn_client.resources.get(api_version="v1", kind="RayCluster") try: delete_response = raycluster_client.delete( name=cluster_name, namespace=namespace diff --git a/gateway/tests/api/test_ray.py b/gateway/tests/api/test_ray.py index c14b22d80..1c0f28220 100644 --- a/gateway/tests/api/test_ray.py +++ b/gateway/tests/api/test_ray.py @@ -78,9 +78,7 @@ def test_kill_cluster(self): success = kill_ray_cluster("some_cluster") self.assertTrue(success) - DynamicClient.resources.get.assert_any_call( - api_version="v1", kind="RayCluster" - ) + DynamicClient.resources.get.assert_any_call(api_version="v1", kind="RayCluster") DynamicClient.resources.get.assert_any_call( api_version="v1", kind="Certificate" )