Remove .model.disutility & docs; adjust imports

upstreamed in #13
iiasa · Jul 29, 2024 · 11d60f4 · 11d60f4
1 parent 840890a
commit 11d60f4
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/message_ix_models/model/transport/build.py b/message_ix_models/model/transport/build.py
@@ -5,11 +5,10 @@
 
 from message_ix_models import ScenarioInfo
 from message_ix_models.util import eval_anno, load_private_data
-from message_ix_models.model import bare, build
+from message_ix_models.model import bare, build, disutility
 from message_ix_models.model.structure import get_codes
 from sdmx.model import Annotation, Code
 
-from message_data.model import disutility
 from message_data.model.transport.utils import consumer_groups
 
 
@@ -142,7 +141,7 @@ def main(context, scenario, **options):
     lu = dict(level="useful", unit="km")
     disutility.add(
         scenario,
-        consumer_groups=consumer_groups(),
+        groups=consumer_groups(),
         technologies=generate_set_elements("technology", "LDV"),
         template=Code(
             id="transport {technology} usage",