@@ -51,6 +51,8 @@ def serialize_tasks_only(project, domain, pkgs, version, folder=None):
5151 fname_index = str (i ).zfill (zero_padded_length )
5252 fname = '{}_{}.pb' .format (fname_index , entity ._id .name )
5353 click .echo (' Writing {} to\n {}' .format (entity ._id , fname ))
54+ if folder :
55+ fname = _os .path .join (folder , fname )
5456 _write_proto_to_file (serialized , fname )
5557
5658 identifier_fname = '{}_{}.identifier.pb' .format (fname_index , entity ._id .name )
@@ -62,7 +64,7 @@ def serialize_tasks_only(project, domain, pkgs, version, folder=None):
6264@system_entry_point
6365def serialize_all (project , domain , pkgs , version , folder = None ):
6466 """
65- In order to register, we have to comply with Admin's endpoints. Those endpoints take the following object . These
67+ In order to register, we have to comply with Admin's endpoints. Those endpoints take the following objects . These
6668 flyteidl.admin.launch_plan_pb2.LaunchPlanSpec
6769 flyteidl.admin.workflow_pb2.WorkflowSpec
6870 flyteidl.admin.task_pb2.TaskSpec
@@ -105,6 +107,8 @@ def serialize_all(project, domain, pkgs, version, folder=None):
105107 fname_index = str (i ).zfill (zero_padded_length )
106108 fname = '{}_{}.pb' .format (fname_index , entity ._id .name )
107109 click .echo (' Writing {} to\n {}' .format (entity ._id , fname ))
110+ if folder :
111+ fname = _os .path .join (folder , fname )
108112 _write_proto_to_file (serialized , fname )
109113
110114 # Not everything serialized will necessarily have an identifier field in it, even though some do (like the
0 commit comments