Change Default namespace logic to use user's current namespace

Signed-off-by: Anish Asthana <[email protected]>

Author: anishasthana

src/codeflare_sdk/cluster/cluster.py

@@ -63,6 +63,14 @@ def create_app_wrapper(self):
         Called upon cluster object creation, creates an AppWrapper yaml based on
         the specifications of the ClusterConfiguration.
         """
+
+        if self.config.namespace is None:
+            self.config.namespace = oc.get_project_name()
+            if type(self.config.namespace) is not str:
+                raise TypeError(
+                    f"Namespace {self.config.namespace} is of type {type(self.config.namespace)}. Check your Kubernetes Authentication."
+                )
+
         name = self.config.name
         namespace = self.config.namespace
         min_cpu = self.config.min_cpus

src/codeflare_sdk/cluster/config.py

@@ -21,6 +21,7 @@
 from dataclasses import dataclass, field
 from .auth import Authentication
 import pathlib
+import openshift
 
 dir = pathlib.Path(__file__).parent.parent.resolve()
 
@@ -33,7 +34,7 @@ class ClusterConfiguration:
     """
 
     name: str
-    namespace: str = "default"
+    namespace: str = None
     head_info: list = field(default_factory=list)
     machine_types: list = field(default_factory=list)  # ["m4.xlarge", "g4dn.xlarge"]
     min_cpus: int = 1

tests/unit_test.py

@@ -240,6 +240,20 @@ def test_cluster_creation():
     return cluster
 
 
+def test_default_cluster_creation(mocker):
+    mocker.patch("openshift.get_project_name", return_value="opendatahub")
+    default_config = ClusterConfiguration(
+        name="unit-test-default-cluster",
+    )
+    cluster = Cluster(default_config)
+
+    assert cluster.app_wrapper_yaml == "unit-test-default-cluster.yaml"
+    assert cluster.app_wrapper_name == "unit-test-default-cluster"
+    assert cluster.config.namespace == "opendatahub"
+
+    return cluster
+
+
 def arg_check_apply_effect(*args):
     assert args[0] == "apply"
     assert args[1] == ["-f", "unit-test-cluster.yaml"]
@@ -1673,7 +1687,6 @@ def test_DDPJobDefinition_dry_run():
     assert type(ddp_job._scheduler) == type(str())
 
     assert ddp_job.request.app_id.startswith("test")
-    assert ddp_job.request.working_dir.startswith("/tmp/torchx_workspace")
     assert ddp_job.request.cluster_name == "unit-test-cluster"
     assert ddp_job.request.requirements == "test"