Review feedback applied, auto-select

Author: Maxusmusti

src/codeflare_sdk/cluster/cluster.py

@@ -549,7 +549,7 @@ def get_current_namespace():  # pragma: no cover
             return None
 
 
-def get_cluster(cluster_name: str, namespace: str = "default", mcad=True):
+def get_cluster(cluster_name: str, namespace: str = "default"):
     try:
         config_check()
         api_instance = client.CustomObjectsApi(api_config_handler())
@@ -564,13 +564,33 @@ def get_cluster(cluster_name: str, namespace: str = "default", mcad=True):
 
     for rc in rcs["items"]:
         if rc["metadata"]["name"] == cluster_name:
+            mcad = _check_aw_exists(cluster_name, namespace)
             return Cluster.from_k8_cluster_object(rc, mcad=mcad)
     raise FileNotFoundError(
         f"Cluster {cluster_name} is not found in {namespace} namespace"
     )
 
 
 # private methods
+def _check_aw_exists(name: str, namespace: str) -> bool:
+    try:
+        config_check()
+        api_instance = client.CustomObjectsApi(api_config_handler())
+        aws = api_instance.list_namespaced_custom_object(
+            group="workload.codeflare.dev",
+            version="v1beta1",
+            namespace=namespace,
+            plural="appwrappers",
+        )
+    except Exception as e:  # pragma: no cover
+        return _kube_api_error_handling(e, print_error=False)
+
+    for aw in aws["items"]:
+        if aw["metadata"]["name"] == name:
+            return True
+    return False
+
+
 def _get_ingress_domain():
     try:
         config_check()

src/codeflare_sdk/utils/kube_api_helpers.py

@@ -22,7 +22,9 @@
 
 
 # private methods
-def _kube_api_error_handling(e: Exception):  # pragma: no cover
+def _kube_api_error_handling(
+    e: Exception, print_error: bool = True
+):  # pragma: no cover
     perm_msg = (
         "Action not permitted, have you put in correct/up-to-date auth credentials?"
     )
@@ -31,11 +33,13 @@ def _kube_api_error_handling(e: Exception):  # pragma: no cover
     if type(e) == config.ConfigException:
         raise PermissionError(perm_msg)
     if type(e) == executing.executing.NotOneValueFound:
-        print(nf_msg)
+        if print_error:
+            print(nf_msg)
         return
     if type(e) == client.ApiException:
         if e.reason == "Not Found":
-            print(nf_msg)
+            if print_error:
+                print(nf_msg)
             return
         elif e.reason == "Unauthorized" or e.reason == "Forbidden":
             raise PermissionError(perm_msg)

tests/test-case-no-mcad.yamls

@@ -51,7 +51,7 @@ spec:
             value: /home/ray/workspace/tls/server.key
           - name: RAY_TLS_CA_CERT
             value: /home/ray/workspace/tls/ca.crt
-          image: quay.io/project-codeflare/ray:2.5.0-py38-cu116
+          image: quay.io/project-codeflare/ray:latest-py39-cu118
           imagePullPolicy: Always
           lifecycle:
             preStop:
@@ -79,7 +79,7 @@ spec:
               nvidia.com/gpu: 0
         imagePullSecrets:
         - name: unit-test-pull-secret
-  rayVersion: 2.5.0
+  rayVersion: 2.7.0
   workerGroupSpecs:
   - groupName: small-group-unit-test-cluster-ray
     maxReplicas: 2
@@ -118,7 +118,7 @@ spec:
             value: /home/ray/workspace/tls/server.key
           - name: RAY_TLS_CA_CERT
             value: /home/ray/workspace/tls/ca.crt
-          image: quay.io/project-codeflare/ray:2.5.0-py38-cu116
+          image: quay.io/project-codeflare/ray:latest-py39-cu118
           lifecycle:
             preStop:
               exec: