From b36ffd956a947f1aa3a689698177fecb9dfd8e84 Mon Sep 17 00:00:00 2001
From: Noe Brucy <noe.brucy@cea.fr>
Date: Tue, 8 Jun 2021 16:05:41 +0200
Subject: [PATCH] [run_selector] better handling of no snapshot

---
 run_selector.py | 13 ++++++++-----
 1 file changed, 8 insertions(+), 5 deletions(-)

diff --git a/run_selector.py b/run_selector.py
index 362570d..ea27e7d 100644
--- a/run_selector.py
+++ b/run_selector.py
@@ -96,9 +96,6 @@ class RunSelector:
         self.namelist = {}
         self.runs = self.get_runs(in_runs, filter_name, filter_nml, sort_run_by)
 
-        if len(self.runs) == 0:
-            raise ValueError("No runs found")
-
         self.info = {}
         for run in self.runs:
             self.info[run] = {}
@@ -119,11 +116,17 @@ class RunSelector:
                 time,
             )
 
-        for i, run in enumerate(self.runs):
+        i = 0
+        for run in self.runs.copy():
             if len(self.nums[run]) == 0:
                 print(f"[WARNING] No snapshot found for run {run}")
                 del self.runs[i]
                 del self.nums[run]
+            else:
+                i += 1
+
+        if len(self.runs) == 0:
+            raise ValueError("No runs found")
 
     def select(
         self,
@@ -328,7 +331,7 @@ class RunSelector:
         if time_max is not None:
             nums = list(filter(lambda n: self.info[run][n]["time"] <= time_max, nums))
 
-        if time is not None:
+        if time is not None and len(nums) > 0:
             filtered_nums = []
             if not isinstance(time, list):
                 time = [time]