From 3ef9319fa7ff426011be75bb8a72aaa35af7fce2 Mon Sep 17 00:00:00 2001
From: Sofia Faltenbacher <faltenbacher.sofia@gmail.com>
Date: Tue, 14 Nov 2023 11:32:09 +0100
Subject: [PATCH] feat(sample_generator): added variable name to metadata dict

---
 causy/sample_generator.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/causy/sample_generator.py b/causy/sample_generator.py
index f2dc531..e315464 100644
--- 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):