Fixed Error in load_stations

run_with_oversampling.py

@@ -2,7 +2,7 @@ __author__ = "Lukas Leufen"
 __date__ = '2020-06-29'
 
 import argparse
-from mlair.workflows import DefaultWorkflow
+from mlair.workflows import DefaultWorkflowHPC
 from mlair.helpers import remove_items
 from mlair.configuration.defaults import DEFAULT_PLOT_LIST
 from mlair.model_modules.model_class import IntelliO3_ts_architecture
@@ -24,7 +24,7 @@ def load_stations(external_station_list=None):
 
 def main(parser_args):
     plots = remove_items(DEFAULT_PLOT_LIST, "PlotConditionalQuantiles")
-    workflow = DefaultWorkflow(stations=load_stations('supplement/German_background_stations.json'),
+    workflow = DefaultWorkflowHPC(stations=load_stations('supplement/German_background_stations.json'),
         #stations=["DEBW013", "DEBW087", "DEBW107", "DEBW076"],
         epochs=150,
         train_model=True, create_new_model=True, network="UBA",

run_without_oversampling.py

@@ -2,14 +2,14 @@ __author__ = "Lukas Leufen"
 __date__ = '2020-06-29'
 
 import argparse
-from mlair.workflows import DefaultWorkflow
+from mlair.workflows import DefaultWorkflowHPC
 from mlair.helpers import remove_items
 from mlair.configuration.defaults import DEFAULT_PLOT_LIST
 from mlair.model_modules.model_class import IntelliO3_ts_architecture
 import os
 
 
-def load_stations(external_station_list = None):
+def load_stations(external_station_list=None):
     import json
     if external_station_list is None:
         external_station_list = 'supplement/station_list_north_german_plain_rural.json'
@@ -24,7 +24,7 @@ def load_stations(external_station_list = None):
 
 def main(parser_args):
     plots = remove_items(DEFAULT_PLOT_LIST, "PlotConditionalQuantiles")
-    workflow = DefaultWorkflow(stations=load_stations('supplement/German_background_stations.json'),
+    workflow = DefaultWorkflowHPC(stations=load_stations('supplement/German_background_stations.json'),
         #stations=["DEBW013", "DEBW087", "DEBW107", "DEBW076"],
         epochs=150,
         train_model=True, create_new_model=True, network="UBA",