Skip to content

Commit

Permalink
Merge pull request #55 from blumenstiel/main
Browse files Browse the repository at this point in the history
Fix kfp yaml name
  • Loading branch information
blumenstiel authored Mar 28, 2024
2 parents dba1fe8 + e43f8f8 commit 46cee2e
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions src/c3/create_operator.py
Original file line number Diff line number Diff line change
Expand Up @@ -60,17 +60,17 @@ def create_dockerfile(dockerfile_template, requirements, target_code, target_dir
def create_kfp_component(name, description, repository, version, command, target_code, target_dir, file_path, inputs, outputs):

inputs_list = str()
for name, options in inputs.items():
inputs_list += f'- {{name: {name}, type: {options["type"]}, description: "{options["description"]}"'
for input, options in inputs.items():
inputs_list += f'- {{name: {input}, type: {options["type"]}, description: "{options["description"]}"'
if options['default'] is not None:
if not options["default"].startswith('"'):
options["default"] = f'"{options["default"]}"'
inputs_list += f', default: {options["default"]}'
inputs_list += '}\n'

outputs_list = str()
for name, options in outputs.items():
outputs_list += f'- {{name: {name}, type: String, description: "{options["description"]}"}}\n'
for output, options in outputs.items():
outputs_list += f'- {{name: {output}, type: String, description: "{options["description"]}"}}\n'

parameter_list = str()
for index, key in enumerate(list(inputs.keys()) + list(outputs.keys())):
Expand Down

0 comments on commit 46cee2e

Please sign in to comment.