Load nodes after project.run

tests/integration/test_project.py

@@ -584,3 +584,16 @@ def test_magic_names(proj_path):
     zntrack.from_rev(node02.name).outs == "Dolor Sit"
     zntrack.from_rev(node03.name).outs == "Test01"
     zntrack.from_rev(grp_node03.name).outs == "Test02"
+
+
+def test_project_run_load(proj_path):
+    """Ensure Nodes are loaded after project is run from the same script."""
+
+    project = zntrack.Project()
+
+    with project:
+        node = zntrack.examples.ParamsToOuts(params="Lorem Ipsum")
+
+    project.run()
+
+    assert node.outs == "Lorem Ipsum"

zntrack/project/zntrack_project.py

@@ -304,7 +304,12 @@ def run(
                 node.save(results=False)
         if not eager and repro:
             self.repro()
-            # TODO should we load the nodes here? Maybe, if lazy loading is implemented.
+            # load the nodes
+            for node_uuid in self.graph.get_sorted_nodes():
+                node = self.graph.nodes[node_uuid]["value"]
+                if node_names is not None and node.name not in node_names:
+                    continue
+                node.load()
 
         if auto_remove:
             self.auto_remove()