From b1a202c41392267ce4122b27de5493b4b8c6568c Mon Sep 17 00:00:00 2001 From: Harrison Cook Date: Tue, 19 Nov 2024 14:13:11 +0000 Subject: [PATCH] Remove assertion limiting datetimes to 0,6,12,18 (#62) * Remove assertion limiting datetimes to 0,6,12,18 --- CITATION.cff | 4 ++++ src/ai_models/model.py | 4 ++-- 2 files changed, 6 insertions(+), 2 deletions(-) diff --git a/CITATION.cff b/CITATION.cff index d362247..a2113c1 100644 --- 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 index 01eda32..c87f78f 100644 --- 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,