Adjust imports for code upstreamed in #13

iiasa · Jul 29, 2024 · e242f8b · e242f8b
1 parent 11d60f4
commit e242f8b
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/message_ix_models/model/transport/cli.py b/message_ix_models/model/transport/cli.py
@@ -92,7 +92,7 @@ def migrate(context, version, check_base, parse, region, source_path, dest):
 @click.pass_obj
 def build_cmd(context, dest, **options):
     """Prepare the model."""
-    from message_data.model import bare
+    from message_ix_models.model import bare
     from message_data.model.transport import build
 
     # Handle --regions; use a sensible default for MESSAGEix-Transport

diff --git a/message_ix_models/model/transport/data/__init__.py b/message_ix_models/model/transport/data/__init__.py
@@ -5,17 +5,17 @@
 import pandas as pd
 from message_ix import make_df
 from message_ix_models import ScenarioInfo
-from message_ix_models.util import add_par_data
-
-from message_data.model.transport.utils import add_commodity_and_level
-from message_data.tools import (
+from message_ix_models.util import (
+    add_par_data,
     broadcast,
     make_io,
     make_matched_dfs,
     make_source_tech,
     same_node,
 )
 
+from message_data.model.transport.utils import add_commodity_and_level
+
 from .groups import get_consumer_groups  # noqa: F401
 from .ldv import get_ldv_data
 from .non_ldv import get_non_ldv_data