fix type hints (#260)

zincware · Mar 21, 2022 · c5e3a96 · c5e3a96
1 parent 3e77134
commit c5e3a96
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/zntrack/zn/dependencies.py b/zntrack/zn/dependencies.py
@@ -2,7 +2,7 @@
 
 import dataclasses
 import pathlib
-from typing import TYPE_CHECKING, List
+from typing import TYPE_CHECKING, List, Union
 
 from zntrack.utils import utils
 
@@ -19,7 +19,7 @@ class NodeAttribute:
     affected_files: List[pathlib.Path]
 
 
-def getdeps(node: Node, attribute: str) -> NodeAttribute:
+def getdeps(node: Union[Node, type(Node)], attribute: str) -> NodeAttribute:
     """Allow for Node attributes as dependencies
 
     Parameters