From e1c7dfc44cb7066b715a9e5cfc039571d65693d5 Mon Sep 17 00:00:00 2001
From: Michael <m.langguth@fz-juelich.de>
Date: Tue, 26 May 2020 12:08:06 +0200
Subject: [PATCH] Make use of get_stat_vars in normalization-class.
---
video_prediction/datasets/era5_dataset_v2.py | 4 +++-
1 file changed, 3 insertions(+), 1 deletion(-)
diff --git a/video_prediction/datasets/era5_dataset_v2.py b/video_prediction/datasets/era5_dataset_v2.py
index 06688190..f25fbc13 100644
--- a/video_prediction/datasets/era5_dataset_v2.py
+++ b/video_prediction/datasets/era5_dataset_v2.py
@@ -15,6 +15,7 @@ from os import path
import sys
sys.path.append(path.abspath('../../workflow_parallel_frame_prediction/'))
from DataPreprocess.process_netCDF_v2 import get_unique_vars
+from DataPreprocess.process_netCDF_v2 import calc_data_stat.get_stat_vars
#from base_dataset import VarLenFeatureVideoDataset
from collections import OrderedDict
from tensorflow.contrib.training import HParams
@@ -203,7 +204,8 @@ class norm_data:
# create all attributes for the instance
for varname in self.varnames:
for stat_name in self.known_norms[norm]:
- setattr(self,varname+stat_name,stat_dict[varname][0][stat_name])
+ #setattr(self,varname+stat_name,stat_dict[varname][0][stat_name])
+ setattr(self,varname+stat_name,get_stat_vars(stat_dict,stat_name,varname))
self.status_ok = True # set status for normalization -> ready
--
GitLab