feat(sample_generator): added variable name to metadata dict

causy-dev · Nov 14, 2023 · 3ef9319 · 3ef9319
1 parent 9eafbb6
commit 3ef9319
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/causy/sample_generator.py b/causy/sample_generator.py
@@ -227,7 +227,10 @@ def generate(self, size):
             output[i] = internal_repr[i].to_list()
             for t in range(size):
                 graph.add_node(
-                    f"{i} - t{t}", [output[i][t]], id_=f"{i}-t{t}", metadata={"time": t}
+                    f"{i} - t{t}",
+                    [output[i][t]],
+                    id_=f"{i}-t{t}",
+                    metadata={"time": t, "variable": i},
                 )
 
         for t in range(1, size):