From 0d1ad0f60510743a4a55f04a92af7e4b91c7bb0e Mon Sep 17 00:00:00 2001
From: Stephan Seitz <stephan.seitz@fau.de>
Date: Wed, 4 Dec 2019 15:49:02 +0100
Subject: [PATCH] Also add TimeloopRuns

---
 pystencils/datahandling/graph_datahandling.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/pystencils/datahandling/graph_datahandling.py b/pystencils/datahandling/graph_datahandling.py
index 4c9b635b7..5d02946a3 100644
--- a/pystencils/datahandling/graph_datahandling.py
+++ b/pystencils/datahandling/graph_datahandling.py
@@ -110,6 +110,7 @@ class GraphDataHandling(pystencils.datahandling.SerialDataHandling):
                 argument_list = [argument_list]
 
         def run(self, time_steps=1):
+            self.parent.call_queue.append(TimeloopRun(self, time_steps))
             super().run(time_steps)
 
     def __init__(self, *args, **kwargs):
-- 
GitLab