test post_load

tests/integration/test_post_load.py

@@ -0,0 +1,25 @@
+import zntrack
+
+
+class NodeWithPostLoad(zntrack.Node):
+    params: float = zntrack.params()
+    outs: float = zntrack.outs()
+
+    def _post_load_(self):
+        self.params += 1
+
+    def run(self):
+        self.outs = self.params
+
+
+def test_post_load(proj_path):
+    project = zntrack.Project()
+    with project:
+        node = NodeWithPostLoad(params=1)
+
+    assert node.params == 1
+    project.repro()
+
+    n = node.from_rev()
+    assert n.params == 2 # modified by _post_load_
+    assert n.outs == 1

zntrack/node.py

@@ -144,7 +144,7 @@ def from_rev(
                 _ = getattr(instance, field.name)
 
         instance._external_ = True
-        if hasattr(instance, "_post_load_"):
+        if not running and hasattr(instance, "_post_load_"):
             with contextlib.suppress(NotImplementedError):
                 instance._post_load_()