Skip to content

Commit 59e65f0

Browse files
committed
Move .transport…ExogenousDataFile to .util.sdmx.Dataflow
- Move and adjust tests. - Sort entries in .transport.files. - Add .transport.files.iter_files(). - Add .transport.key.exo, populated using iter_files() - Adjust references.
1 parent da8f2c6 commit 59e65f0

File tree

14 files changed

+486
-447
lines changed

14 files changed

+486
-447
lines changed

message_ix_models/model/transport/build.py

+4-3
Original file line numberDiff line numberDiff line change
@@ -153,13 +153,14 @@ def add_exogenous_data(c: Computer, info: ScenarioInfo) -> None:
153153
import message_ix_models.tools.iea.web # noqa: F401
154154
from message_ix_models.project.ssp import SSP_2017, SSP_2024
155155
from message_ix_models.tools.exo_data import prepare_computer
156+
from message_ix_models.util.sdmx import Dataflow
156157

157158
# Ensure that the MERtoPPP data provider is available
158159
from . import (
159160
data, # noqa: F401
160161
key,
161162
)
162-
from .files import FILES, add
163+
from .files import iter_files
163164

164165
# Added keys
165166
keys = {}
@@ -243,15 +244,15 @@ def add_exogenous_data(c: Computer, info: ScenarioInfo) -> None:
243244
# Data from files
244245

245246
# Identify the mode-share file according to the config setting
246-
add(
247+
Dataflow(
247248
key="mode share:n-t:exo",
248249
path=("mode-share", config.mode_share),
249250
name="Reference (base year) mode share",
250251
units="dimensionless",
251252
replace=True,
252253
)
253254

254-
for f in FILES:
255+
for _, f in iter_files():
255256
c.add("", f, context=context)
256257

257258

message_ix_models/model/transport/demand.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -13,10 +13,10 @@
1313
from message_ix_models.report.key import GDP
1414
from message_ix_models.util import broadcast
1515

16-
from . import files as exo
1716
from .key import (
1817
cg,
1918
cost,
19+
exo,
2020
fv,
2121
fv_cny,
2222
gdp_cap,

message_ix_models/model/transport/disutility.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22

33
from genno import Key, Quantity, quote
44

5-
from . import files as exo
5+
from .key import exo
66

77
if TYPE_CHECKING:
88
from genno import Computer

0 commit comments

Comments
 (0)