Fix pipeline component returning a list of length 1

This commit is contained in:
allegroai 2022-08-09 09:45:19 +03:00
parent 6924f32d6c
commit e090995ad8

View File

@ -497,7 +497,7 @@ if __name__ == '__main__':
results = {function_name}(**kwargs)
result_names = {function_return}
if result_names:
if not isinstance(results, (tuple, list)) or (len(result_names) == 1 and len(results) != 1):
if not isinstance(results, (tuple, list)) or len(result_names) == 1:
results = [results]
for name, artifact in zip(result_names, results):
task.upload_artifact(name=name, artifact_object=artifact)