added if statement for if labels are None

utils/data_logging_utils.py

@@ -71,7 +71,10 @@ class LogWriter():
 
         self.current_iteration = 1
 
-        self.labels = self.labels_generator(labels)
+        if labels is not None:
+            self.labels = self.labels_generator(labels)
+        else:
+            pass
 
         self.logger = logging.getLogger()
         file_handler = logging.FileHandler(