From b36b9678a2bf1a19213563b239b27addd6b72ff5 Mon Sep 17 00:00:00 2001
From: Michael <m.langguth@fz-juelich.de>
Date: Fri, 28 May 2021 15:56:51 +0200
Subject: [PATCH] Add call of plot_conditional_quantiles to
 main_visualize_postprocess.py.

---
 .../main_scripts/main_visualize_postprocess.py               | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/video_prediction_tools/main_scripts/main_visualize_postprocess.py b/video_prediction_tools/main_scripts/main_visualize_postprocess.py
index 1a8e75e1..4658dc1a 100644
--- a/video_prediction_tools/main_scripts/main_visualize_postprocess.py
+++ b/video_prediction_tools/main_scripts/main_visualize_postprocess.py
@@ -774,9 +774,7 @@ class Postprocess(TrainModel):
         # release some memory
         Postprocess.clean_obj_attribute(self, "eval_metrics_ds")
 
-
-        cond_quantile_vars = ["{0}_{1}_fcst".format(self.vars_in[self.channel], self.model),
-                              "{0}_ref".format(self.vars_in[self.channel])]
+        # the variables for conditional quantile plot
         var_fcst = "{0}_{1}_fcst".format(self.vars_in[self.channel], self.model)
         var_ref = "{0}_ref".format(self.vars_in[self.channel])
 
@@ -1178,6 +1176,7 @@ def main():
     postproc_instance.run()
     postproc_instance.handle_eval_metrics()
     postproc_instance.plot_example_forecasts(metric=args.eval_metrics[0], channel=args.channel)
+    postproc_instance.plot_conditional_quantiles()
 
 
 if __name__ == '__main__':
-- 
GitLab