diff --git a/test/e2e/v1beta1/scripts/gh-actions/run-e2e-tune-api.py b/test/e2e/v1beta1/scripts/gh-actions/run-e2e-tune-api.py index 458420dfe42..b2f55cfeaf2 100644 --- a/test/e2e/v1beta1/scripts/gh-actions/run-e2e-tune-api.py +++ b/test/e2e/v1beta1/scripts/gh-actions/run-e2e-tune-api.py @@ -3,6 +3,7 @@ from kubeflow.katib import KatibClient from kubeflow.katib import search +from kubernetes import client from verify import verify_experiment_results # Experiment timeout is 40 min. @@ -71,6 +72,11 @@ def objective(parameters): katib_client = KatibClient() + namespace_labels = client.CoreV1Api().read_namespace(args.namespace).metadata.labels + if 'katib.kubeflow.org/metrics-collector-injection' not in namespace_labels: + namespace_labels['katib.kubeflow.org/metrics-collector-injection'] = 'enabled' + client.CoreV1Api().patch_namespace(args.namespace, {'metadata': {'labels': namespace_labels}}) + # Test with run_e2e_experiment_create_by_tune exp_name = "tune-example" exp_namespace = args.namespace