From ada3420ee50d51a8a109e0ce665418c053be9e54 Mon Sep 17 00:00:00 2001
From: Michael <m.langguth@fz-juelich.de>
Date: Tue, 18 May 2021 14:22:02 +0200
Subject: [PATCH] Limit maximum index in populate_eval_metric_ds-method of
main_visualize_postprocess.py.
---
.../main_scripts/main_visualize_postprocess.py | 3 ++-
1 file changed, 2 insertions(+), 1 deletion(-)
diff --git a/video_prediction_tools/main_scripts/main_visualize_postprocess.py b/video_prediction_tools/main_scripts/main_visualize_postprocess.py
index 196cfee6..00f4991a 100644
--- a/video_prediction_tools/main_scripts/main_visualize_postprocess.py
+++ b/video_prediction_tools/main_scripts/main_visualize_postprocess.py
@@ -542,8 +542,9 @@ class Postprocess(TrainModel):
varname_ref = "{0}_ref".format(varname)
# reset init-time coordinate of metric_ds in place
+ ind_end = ind_start + min(self.batch_size, len(data_ds["init_time"]))
init_times_metric = metric_ds["init_time"].values
- init_times_metric[ind_start:ind_start+self.batch_size] = data_ds["init_time"]
+ init_times_metric[ind_start:ind_end] = data_ds["init_time"]
metric_ds = metric_ds.assign_coords(init_time=init_times_metric)
# populate metric_ds
for fcst_prod in self.fcst_products.keys():
--
GitLab