diff --git a/sdk/python/kfp/_client.py b/sdk/python/kfp/_client.py index b4bb8222681..ccc73a20309 100644 --- a/sdk/python/kfp/_client.py +++ b/sdk/python/kfp/_client.py @@ -21,6 +21,7 @@ import yaml from datetime import datetime +from .compiler import compiler class Client(object): """ API Client for KubeFlow Pipeline. @@ -138,7 +139,8 @@ def run_pipeline(self, experiment_id, job_name, pipeline_package_path, params={} pipeline_obj = self._extract_pipeline_yaml(pipeline_package_path) pipeline_json_string = json.dumps(pipeline_obj) - api_params = [kfp_run.ApiParameter(name=k, value=str(v)) for k,v in params.items()] + api_params = [kfp_run.ApiParameter(name=compiler.Compiler()._sanitize_name(k), value=str(v)) + for k,v in params.items()] key = kfp_run.models.ApiResourceKey(id=experiment_id, type=kfp_run.models.ApiResourceType.EXPERIMENT) reference = kfp_run.models.ApiResourceReference(key, kfp_run.models.ApiRelationship.OWNER)