From a83b64b0dcf8def2e3d028107a310244c4871120 Mon Sep 17 00:00:00 2001
From: lukas leufen <l.leufen@fz-juelich.de>
Date: Mon, 22 Jun 2020 10:28:41 +0200
Subject: [PATCH] include epochs in src/run module

---
 src/run.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/run.py b/src/run.py
index eda0373c..11029817 100644
--- a/src/run.py
+++ b/src/run.py
@@ -26,7 +26,8 @@ def run(stations=['DEBW107', 'DEBY081', 'DEBW013', 'DEBW076', 'DEBW087', 'DEBW00
         evaluate_bootstraps=True, number_of_bootstraps=None, create_new_bootstraps=False,
         plot_list=None,
         model=None,
-        batch_size=None):
+        batch_size=None,
+        epochs=None):
 
     params = inspect.getfullargspec(ExperimentSetup).args
     kwargs = {k: v for k, v in locals().items() if k in params}
-- 
GitLab