Skip to content

Commit dc684b3

Browse files
Remove openshift_oauth configuration
1 parent 3fd0905 commit dc684b3

10 files changed

+10
-35
lines changed

Diff for: src/codeflare_sdk/cluster/cluster.py

+3-18
Original file line numberDiff line numberDiff line change
@@ -81,21 +81,18 @@ def _client_headers(self):
8181
)
8282
}
8383

84-
@property
85-
def _client_verify_tls(self):
86-
return not self.config.openshift_oauth
8784

8885
@property
8986
def job_client(self):
9087
k8client = api_config_handler() or client.ApiClient()
9188
if self._job_submission_client:
9289
return self._job_submission_client
93-
if self.config.openshift_oauth:
90+
if is_openshift_cluster():
9491
print(k8client.configuration.get_api_key_with_prefix("authorization"))
9592
self._job_submission_client = JobSubmissionClient(
9693
self.cluster_dashboard_uri(),
9794
headers=self._client_headers,
98-
verify=self._client_verify_tls,
95+
verify=False,
9996
)
10097
else:
10198
self._job_submission_client = JobSubmissionClient(
@@ -210,7 +207,6 @@ def create_app_wrapper(self):
210207
image_pull_secrets=image_pull_secrets,
211208
dispatch_priority=dispatch_priority,
212209
priority_val=priority_val,
213-
openshift_oauth=self.config.openshift_oauth,
214210
ingress_domain=ingress_domain,
215211
ingress_options=ingress_options,
216212
write_to_file=write_to_file,
@@ -351,7 +347,7 @@ def is_dashboard_ready(self) -> bool:
351347
self.cluster_dashboard_uri(),
352348
headers=self._client_headers,
353349
timeout=5,
354-
verify=self._client_verify_tls,
350+
verify=False,
355351
)
356352
except requests.exceptions.SSLError: # pragma no cover
357353
# SSL exception occurs when oauth ingress has been created but cluster is not up
@@ -491,23 +487,13 @@ def from_k8_cluster_object(
491487
rc, mcad=True, ingress_domain=None, ingress_options={}, write_to_file=False
492488
):
493489
config_check()
494-
openshift_oauth = False
495490
if (
496491
rc["metadata"]["annotations"]["sdk.codeflare.dev/local_interactive"]
497492
== "True"
498493
):
499494
local_interactive = True
500495
else:
501496
local_interactive = False
502-
if "codeflare.dev/oauth" in rc["metadata"]["annotations"]:
503-
openshift_oauth = (
504-
rc["metadata"]["annotations"]["codeflare.dev/oauth"] == "True"
505-
)
506-
else:
507-
for container in rc["spec"]["headGroupSpec"]["template"]["spec"][
508-
"containers"
509-
]:
510-
openshift_oauth = "oauth-proxy" in container["name"]
511497
machine_types = (
512498
rc["metadata"]["labels"]["orderedinstance"].split("_")
513499
if "orderedinstance" in rc["metadata"]["labels"]
@@ -558,7 +544,6 @@ def from_k8_cluster_object(
558544
ingress_domain=ingress_domain,
559545
ingress_options=ingress_options,
560546
write_to_file=write_to_file,
561-
openshift_oauth=openshift_oauth,
562547
)
563548
return Cluster(cluster_config)
564549

Diff for: src/codeflare_sdk/cluster/config.py

-1
Original file line numberDiff line numberDiff line change
@@ -52,7 +52,6 @@ class ClusterConfiguration:
5252
local_interactive: bool = False
5353
image_pull_secrets: list = field(default_factory=list)
5454
dispatch_priority: str = None
55-
openshift_oauth: bool = False # NOTE: to use the user must have permission to create a RoleBinding for system:auth-delegator
5655
ingress_options: dict = field(default_factory=dict)
5756
ingress_domain: str = None
5857
write_to_file: bool = False

Diff for: src/codeflare_sdk/templates/base-template.yaml

-1
Original file line numberDiff line numberDiff line change
@@ -42,7 +42,6 @@ spec:
4242
metadata:
4343
annotations:
4444
sdk.codeflare.dev/local_interactive: "False"
45-
codeflare.dev/oauth: 'False'
4645
labels:
4746
workload.codeflare.dev/appwrapper: "aw-kuberay"
4847
controller-tools.k8s.io: "1.0"

Diff for: src/codeflare_sdk/utils/generate_yaml.py

+4-9
Original file line numberDiff line numberDiff line change
@@ -219,17 +219,16 @@ def update_rayclient_ingress(
219219
spec["rules"][0]["host"] = f"rayclient-{cluster_name}-{namespace}.{ingress_domain}"
220220

221221

222-
def update_names(yaml, item, appwrapper_name, cluster_name, namespace, openshift_oauth):
222+
def update_names(yaml, item, appwrapper_name, cluster_name, namespace):
223223
metadata = yaml.get("metadata")
224224
metadata["name"] = appwrapper_name
225225
metadata["namespace"] = namespace
226226
lower_meta = item.get("generictemplate", {}).get("metadata")
227227
lower_meta["labels"]["workload.codeflare.dev/appwrapper"] = appwrapper_name
228-
lower_meta["annotations"]["codeflare.dev/oauth"] = f"{openshift_oauth}"
229228
lower_meta["name"] = cluster_name
230229
lower_meta["namespace"] = namespace
231230
lower_spec = item.get("generictemplate", {}).get("spec")
232-
if openshift_oauth:
231+
if is_openshift_cluster():
233232
cookie_secret_env_var = {
234233
"name": "COOKIE_SECRET",
235234
"valueFrom": {
@@ -600,9 +599,6 @@ def enable_openshift_oauth(user_yaml, cluster_name, namespace):
600599
)
601600
# allows for setting value of Cluster object when initializing object from an existing AppWrapper on cluster
602601
user_yaml["metadata"]["annotations"] = user_yaml["metadata"].get("annotations", {})
603-
user_yaml["metadata"]["annotations"][
604-
"codeflare-sdk-use-oauth"
605-
] = "true" # if the user gets an
606602
ray_headgroup_pod = user_yaml["spec"]["resources"]["GenericItems"][0][
607603
"generictemplate"
608604
]["spec"]["headGroupSpec"]["template"]["spec"]
@@ -711,7 +707,6 @@ def generate_appwrapper(
711707
image_pull_secrets: list,
712708
dispatch_priority: str,
713709
priority_val: int,
714-
openshift_oauth: bool,
715710
ingress_domain: str,
716711
ingress_options: dict,
717712
write_to_file: bool,
@@ -723,7 +718,7 @@ def generate_appwrapper(
723718
ingress_item = resources["resources"].get("GenericItems")[1]
724719
route_item = resources["resources"].get("GenericItems")[2]
725720
update_names(
726-
user_yaml, item, appwrapper_name, cluster_name, namespace, openshift_oauth
721+
user_yaml, item, appwrapper_name, cluster_name, namespace
727722
)
728723
update_labels(user_yaml, instascale, instance_types)
729724
update_priority(user_yaml, item, dispatch_priority, priority_val)
@@ -774,7 +769,7 @@ def generate_appwrapper(
774769

775770
delete_route_or_ingress(resources["resources"])
776771

777-
if openshift_oauth:
772+
if is_openshift_cluster():
778773
enable_openshift_oauth(user_yaml, cluster_name, namespace)
779774

780775
directory_path = os.path.expanduser("~/.codeflare/appwrapper/")

Diff for: tests/e2e/mnist_raycluster_sdk_oauth_test.py

-1
Original file line numberDiff line numberDiff line change
@@ -51,7 +51,6 @@ def run_mnist_raycluster_sdk_oauth(self):
5151
num_gpus=0,
5252
instascale=False,
5353
image=ray_image,
54-
openshift_oauth=True,
5554
write_to_file=True,
5655
)
5756
)

Diff for: tests/test-case-no-mcad.yamls

-1
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,6 @@ apiVersion: ray.io/v1
33
kind: RayCluster
44
metadata:
55
annotations:
6-
codeflare.dev/oauth: 'False'
76
sdk.codeflare.dev/ingress_domain: apps.cluster.awsroute.org
87
sdk.codeflare.dev/local_interactive: 'False'
98
labels:

Diff for: tests/test-case-prio.yaml

-1
Original file line numberDiff line numberDiff line change
@@ -33,7 +33,6 @@ spec:
3333
kind: RayCluster
3434
metadata:
3535
annotations:
36-
codeflare.dev/oauth: 'False'
3736
sdk.codeflare.dev/ingress_domain: apps.cluster.awsroute.org
3837
sdk.codeflare.dev/local_interactive: 'False'
3938
labels:

Diff for: tests/test-case.yaml

-1
Original file line numberDiff line numberDiff line change
@@ -32,7 +32,6 @@ spec:
3232
kind: RayCluster
3333
metadata:
3434
annotations:
35-
codeflare.dev/oauth: 'False'
3635
sdk.codeflare.dev/ingress_domain: apps.cluster.awsroute.org
3736
sdk.codeflare.dev/local_interactive: 'False'
3837
labels:

Diff for: tests/test-default-appwrapper.yaml

-1
Original file line numberDiff line numberDiff line change
@@ -32,7 +32,6 @@ spec:
3232
annotations:
3333
sdk.codeflare.dev/ingress_domain: apps.cluster.awsroute.org
3434
sdk.codeflare.dev/local_interactive: 'False'
35-
codeflare.dev/oauth: 'False'
3635
labels:
3736
controller-tools.k8s.io: '1.0'
3837
workload.codeflare.dev/appwrapper: unit-test-default-cluster

Diff for: tests/unit_test.py

+3-1
Original file line numberDiff line numberDiff line change
@@ -3148,14 +3148,16 @@ def test_gen_app_wrapper_with_oauth(mocker: MockerFixture):
31483148
"codeflare_sdk.cluster.cluster.get_current_namespace",
31493149
return_value="opendatahub",
31503150
)
3151+
mocker.patch(
3152+
"codeflare_sdk.utils.generate_yaml.is_openshift_cluster", return_value=True
3153+
)
31513154
write_user_appwrapper = MagicMock()
31523155
mocker.patch(
31533156
"codeflare_sdk.utils.generate_yaml.write_user_appwrapper", write_user_appwrapper
31543157
)
31553158
Cluster(
31563159
ClusterConfiguration(
31573160
"test_cluster",
3158-
openshift_oauth=True,
31593161
image="quay.io/project-codeflare/ray:latest-py39-cu118",
31603162
ingress_domain="apps.cluster.awsroute.org",
31613163
write_to_file=True,

0 commit comments

Comments
 (0)