From e5a76c50bd9ffe7bc8f02f9ffe7e69330a429061 Mon Sep 17 00:00:00 2001 From: Bradley Jiang Date: Tue, 4 Dec 2018 12:34:32 -0800 Subject: [PATCH] Sanitize inputs from SDK when submitting run. --- sdk/python/kfp/_client.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) 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)