Remove assertion limiting datetimes to 0,6,12,18 (#62)

* Remove assertion limiting datetimes to 0,6,12,18
ecmwf-lab · Nov 19, 2024 · b1a202c · b1a202c
1 parent c5847cd
commit b1a202c
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/CITATION.cff b/CITATION.cff
@@ -20,6 +20,10 @@ authors:
     family-names: Dramsch
     affiliation: ECMWF
     orcid: 'https://orcid.org/0000-0001-8273-905X'
+  - given-names: Harrison
+    family-names: Cook
+    affiliation: ECMWF
+    orcid: 'https://orcid.org/0009-0009-3207-4876'
   - given-names: Matthew
     family-names: Chantry
     affiliation: ECMWF

diff --git a/src/ai_models/model.py b/src/ai_models/model.py
@@ -260,7 +260,7 @@ def _datetimes(self, dates):
         assert isinstance(time, int)
         if time < 100:
             time *= 100
-        assert time in (0, 600, 1200, 1800), time
+        # assert time in (0, 600, 1200, 1800), time
 
         lagged = self.lagged
         if not lagged:
@@ -300,7 +300,7 @@ def datetimes(self, step=0):
         if time < 100:
             time *= 100
 
-        assert time in (0, 600, 1200, 1800), time
+        # assert time in (0, 600, 1200, 1800), time
 
         full = datetime.datetime(
             date // 10000,