Skip to content

Commit 757f782

Browse files
committed
fix: compatible for higher version of sprig
Signed-off-by: zjgemi <[email protected]>
1 parent 221550c commit 757f782

File tree

1 file changed

+7
-6
lines changed

1 file changed

+7
-6
lines changed

src/dflow/step.py

Lines changed: 7 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -722,8 +722,8 @@ def merge_output_artifact(art, parent, layer=0):
722722
self.template.set_slices(self.template.slices)
723723
self.with_param = argo_range(if_expression(
724724
"%s %% %s > 0" % (nslices, group_size),
725-
"%s/%s + 1" % (nslices, group_size),
726-
"%s/%s" % (nslices, group_size)))
725+
"sprig.int(%s/%s) + 1" % (nslices, group_size),
726+
"sprig.int(%s/%s)" % (nslices, group_size)))
727727
elif self.with_param is not None:
728728
self.template.inputs.parameters["dflow_with_param"] = \
729729
InputParameter()
@@ -745,8 +745,8 @@ def merge_output_artifact(art, parent, layer=0):
745745
self.template.set_slices(self.template.slices)
746746
self.with_param = argo_range(if_expression(
747747
"%s %% %s > 0" % (nslices, group_size),
748-
"%s/%s + 1" % (nslices, group_size),
749-
"%s/%s" % (nslices, group_size)))
748+
"sprig.int(%s/%s) + 1" % (nslices, group_size),
749+
"sprig.int(%s/%s)" % (nslices, group_size)))
750750
if self.with_sequence is not None:
751751
self.template.inputs.parameters["dflow_sequence_start"] = \
752752
InputParameter()
@@ -799,8 +799,9 @@ def merge_output_artifact(art, parent, layer=0):
799799
self.with_sequence = argo_sequence(
800800
count=if_expression(
801801
"%s %% %s > 0" % (nslices, group_size),
802-
"%s/%s + 1" % (nslices, group_size),
803-
"%s/%s" % (nslices, group_size)), format=format)
802+
"sprig.int(%s/%s) + 1" % (nslices, group_size),
803+
"sprig.int(%s/%s)" % (nslices, group_size)),
804+
format=format)
804805

805806
self.inputs.parameters["dflow_nslices"] = InputParameter(
806807
value=nslices)

0 commit comments

Comments
 (0)