fix the bugs for integrating weatherbench3DCNN architecture

video_prediction_tools/model_modules/model_architectures.py

@@ -13,7 +13,7 @@ def known_models():
         'convLSTM_gan': "ConvLstmGANVideoPredictionModel",
         'ours_vae_l1': 'SAVPVideoPredictionModel',
         'ours_gan': 'SAVPVideoPredictionModel',
-        "weatherBench": "WeatherBenchModel "
+        "weatherBench": "WeatherBenchModel"
     }
 
     return model_mappings

video_prediction_tools/model_modules/video_prediction/models/weatherBench3DCNN.py

@@ -13,7 +13,7 @@ from model_modules.video_prediction.losses import  *
 
 class WeatherBenchModel(object):
 
-    def __init__(self, hparams_dict=None):
+    def __init__(self, hparams_dict=None,**kwargs):
         """
         This is class for building weahterBench architecture by using updated hparameters
         args:
@@ -67,7 +67,8 @@ class WeatherBenchModel(object):
             loss_fun = "mse",
             shuffle_on_val= True,
             filter = 4,
-            kernels = 4
+            kernels = 4,
+    
         )
         return hparams