more update for supporting monthly data

E3SM-Project · Feb 6, 2024 · 9772f90 · 9772f90
1 parent 102f88d
commit 9772f90
Showing 1 changed file with 19 additions and 1 deletion.
diff --git a/e3sm_diags/driver/utils/dataset.py b/e3sm_diags/driver/utils/dataset.py
@@ -321,7 +321,25 @@ def _find_climo_file(self, path_name, data_name, season):
                 return os.path.join(path_name, filename)
         # The below is only ran on model data, because a shorter name is passed into this software. Won't work when use month name such as '01' as season.
         for filename in dir_files:
-            if season in ["ANN", "DJF", "MAM", "JJA", "SON"]:
+            if season in [
+                "ANN",
+                "DJF",
+                "MAM",
+                "JJA",
+                "SON",
+                "01",
+                "02",
+                "03",
+                "04",
+                "05",
+                "06",
+                "07",
+                "08",
+                "09",
+                "10",
+                "11",
+                "12",
+            ]:
                 if filename.startswith(data_name) and season in filename:
                     return os.path.join(path_name, filename)
         # No file found.