modif. for the case where the evaluated model is in the archive

MeteoSwiss · Jan 5, 2024 · 1e69113 · 1e69113
1 parent b65f6f7
commit 1e69113
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/rainforest/performance/eval_get_estimates.py b/rainforest/performance/eval_get_estimates.py
@@ -213,12 +213,14 @@ def _get_data(self):
 
         # Check if the RF-models are there
         for model in self.modellist:
+            self.model_files = {}
             path = self.qpefolder+'{}'.format(model)
-            if len(os.listdir(path)) == 0 :
+            if not os.path.exists(path) or (len(os.listdir(path)) == 0):
+                logging.info('Extracting {} files from archive'.format(model))
                 try:
                     path = self.qpefolder+'{}'.format(model)
-                    self.ref_files[ref] = retrieve_prod(path + '/', self.tstart, 
-                                                                    self.tend, ref)
+                    self.model_files[model] = retrieve_prod(path + '/', self.tstart, 
+                                                                    self.tend, model)
                     logging.info('Model data: {} taken from file archive!'.format(model))
                 except:
                     logging.error('No QPE maps available for {}, please check path or produce QPE maps'.format(model))