fix #528

zincware · Mar 17, 2023 · 6c48e13 · 6c48e13
1 parent a8c3e35
commit 6c48e13
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/zntrack/fields/zn/__init__.py b/zntrack/fields/zn/__init__.py
@@ -338,9 +338,9 @@ def get_stage_add_argument(self, instance) -> typing.List[tuple]:
 class _SaveNodes(znflow.utils.IterableHandler):
     def default(self, value, **kwargs):
         name = kwargs["name"]
-        if hasattr(value, "save"):
+        if isinstance(value, znflow.Node):
             value.name = name
-            value.save()
+            value.save(results=False)
         return value