added deterministic order of creating nodes

luxonis_train/utils/general.py

@@ -265,20 +265,22 @@ def validate_packet(data: Packet[Tensor], protocol: type[BaseModel]) -> Packet[T
 # TEST:
 def traverse_graph(
     graph: dict[str, list[str]], nodes: dict[str, T]
-) -> Generator[tuple[str, T, list[str], set[str]], None, None]:
+) -> Generator[tuple[str, T, list[str], list[str]], None, None]:
     """Traverses the graph in topological order.
 
     @type graph: dict[str, list[str]]
     @param graph: Graph in a format of a dictionary of predecessors. Keys are node
         names, values are inputs to the node (list of node names).
     @type nodes: dict[str, T]
     @param nodes: Dictionary mapping node names to node objects.
-    @rtype: Generator[tuple[str, T, list[str], set[str]], None, None]
+    @rtype: Generator[tuple[str, T, list[str], list[str]], None, None]
     @return: Generator of tuples containing node name, node object, node dependencies
         and unprocessed nodes.
     @raises RuntimeError: If the graph is malformed.
     """
-    unprocessed_nodes = set(nodes.keys())
+    unprocessed_nodes = sorted(
+        set(nodes.keys())
+    )  # sort the set to allow reproducibility
     processed: set[str] = set()
 
     while unprocessed_nodes: