From 35a520980017f562b5a61d4ec4bbc9961f1009ad Mon Sep 17 00:00:00 2001 From: zjgemi Date: Tue, 16 Sep 2025 17:10:58 +0800 Subject: [PATCH] fix: add an option symlink to OP.set_output Signed-off-by: zjgemi --- src/dflow/python/op.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/dflow/python/op.py b/src/dflow/python/op.py index 3f910669..c09fa1f6 100644 --- a/src/dflow/python/op.py +++ b/src/dflow/python/op.py @@ -398,9 +398,9 @@ def from_graph(cls, graph): op = getattr(mod, graph["name"]) return op - def set_output(self, name, value): + def set_output(self, name, value, symlink=True): self.outputs[name] = value - self.handle_outputs({name: value}, symlink=True) + self.handle_outputs({name: value}, symlink=symlink) def handle_outputs(self, outputs, symlink=False): os.makedirs("%s/outputs/parameters" % self.tmp_root, exist_ok=True)