From 903817aac107049e57d05fb75d8e9eeb641ba17f Mon Sep 17 00:00:00 2001
From: Michael <m.langguth@fz-juelich.de>
Date: Tue, 17 Aug 2021 13:35:41 +0200
Subject: [PATCH] Removal of unused argument from generate_fetches in
 main_train_models.py.

---
 video_prediction_tools/main_scripts/main_train_models.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/video_prediction_tools/main_scripts/main_train_models.py b/video_prediction_tools/main_scripts/main_train_models.py
index e092a855..300291ba 100644
--- a/video_prediction_tools/main_scripts/main_train_models.py
+++ b/video_prediction_tools/main_scripts/main_train_models.py
@@ -405,7 +405,7 @@ class TrainModel(object):
 
         return self.val_fetches
 
-    def generate_fetches(self, fetch_list, nest_element=None):
+    def generate_fetches(self, fetch_list):
         """
         Generates dictionary of fetches from video model instance
         :param fetch_list: list of attributes of video model instance that are of particular interest; 
-- 
GitLab