diff --git a/sdk/python/v1beta1/kubeflow/katib/api/katib_client.py b/sdk/python/v1beta1/kubeflow/katib/api/katib_client.py index d0c4a137e98..61f89056a31 100644 --- a/sdk/python/v1beta1/kubeflow/katib/api/katib_client.py +++ b/sdk/python/v1beta1/kubeflow/katib/api/katib_client.py @@ -110,7 +110,7 @@ def create_experiment( constants.EXPERIMENT_PLURAL, experiment, ) - experiment_name = outputs.metadata.name # if "generate_name" is used, "name" gets a prefix from server + experiment_name = outputs["metadata"]["name"] # if "generate_name" is used, "name" gets a prefix from server except multiprocessing.TimeoutError: raise TimeoutError( f"Timeout to create Katib Experiment: {namespace}/{experiment_name}"