Skip to content

Commit d6ed174

Browse files
fix: broken source imports (#450) (#454)
## Description It looks like a couple of source imports were broken during #430, this fixes the import errors. ***As a contributor to the Anemoi framework, please ensure that your changes include unit tests, updates to any affected dependencies and documentation, and have been tested in a parallel setting (i.e., with multiple GPUs). As a reviewer, you are also responsible for verifying these aspects and requesting changes if they are not adequately addressed. For guidelines about those please refer to https://anemoi.readthedocs.io/en/latest/*** By opening this pull request, I affirm that all authors agree to the [Contributor License Agreement.](https://github.com/ecmwf/codex/blob/main/Legal/contributor_license_agreement.md)
1 parent 5cef720 commit d6ed174

File tree

2 files changed

+3
-4
lines changed

2 files changed

+3
-4
lines changed

src/anemoi/datasets/create/sources/accumulations.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -998,12 +998,12 @@ def _execute(
998998
and request.get("stream", "oper") == "oper"
999999
and request.get("accumulation_period") == 24
10001000
):
1001-
from .accumulations2 import accumulations as accumulations2
1001+
from .accumulations2 import Accumulations2Source
10021002

10031003
LOG.warning(
10041004
"🧪️ Experimental features: Using accumulations2, because class=ea stream=oper and accumulation_period=24"
10051005
)
1006-
return accumulations2(context, dates, **request)
1006+
return Accumulations2Source._execute(context, dates, **request)
10071007

10081008
_to_list(request["param"])
10091009
class_ = request.get("class", "od")

src/anemoi/datasets/create/sources/tendencies.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -110,8 +110,7 @@ def _execute(dates: list[datetime.datetime], time_increment: Any, **kwargs: Any)
110110
shifted_dates = [d - time_increment for d in dates]
111111
all_dates = sorted(list(set(dates + shifted_dates)))
112112

113-
# from .mars import execute as mars
114-
from anemoi.datasets.create.mars import execute as mars
113+
from .mars import mars
115114

116115
ds = mars(dates=all_dates, **kwargs)
117116

0 commit comments

Comments
 (0)