use uuid property

zincware · Jun 27, 2023 · 50a6845 · 50a6845
1 parent 347bca2
commit 50a6845
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/zntrack/core/node.py b/zntrack/core/node.py
@@ -9,7 +9,6 @@
 import pathlib
 import time
 import typing
-import uuid
 
 import dvc.api
 import dvc.cli
@@ -172,7 +171,7 @@ def save(
     ) -> None:
         """Save the node's output to disk."""
         if uuid_only:
-            (self.nwd / "uuid").write_text(str(uuid.uuid4()))
+            (self.nwd / "uuid").write_text(str(self.uuid))
             return
 
         # TODO have an option to save and run dvc commit afterwards.