diff --git a/video_prediction_tools/model_modules/video_prediction/models/__init__.py b/video_prediction_tools/model_modules/video_prediction/models/__init__.py index 10c5c72e932e38597731a61555f56803d639bde6..e066c06ada0a384c6de7363a9bc9379a5d3f4786 100644 --- a/video_prediction_tools/model_modules/video_prediction/models/__init__.py +++ b/video_prediction_tools/model_modules/video_prediction/models/__init__.py @@ -7,7 +7,7 @@ from .vanilla_convLSTM_model import VanillaConvLstmVideoPredictionModel from .test_model import TestModelVideoPredictionModel from model_modules.model_architectures import known_models from .convLSTM_GAN_model import ConvLstmGANVideoPredictionModel -from .weatherBench3DCNN import WeatherBenchModel +from .weatherBench3Dcnn import WeatherBenchModel def get_model_class(model): model_mappings = known_models()