From 09e4a652d5aeac185cdb0e3587a947609ca401d1 Mon Sep 17 00:00:00 2001 From: Measrainsey Meng Date: Wed, 11 Sep 2024 14:43:14 +0200 Subject: [PATCH] Add cooling variant to type hints --- message_ix_models/tools/costs/regional_differentiation.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/message_ix_models/tools/costs/regional_differentiation.py b/message_ix_models/tools/costs/regional_differentiation.py index abb764bf6f..da071ab32e 100644 --- a/message_ix_models/tools/costs/regional_differentiation.py +++ b/message_ix_models/tools/costs/regional_differentiation.py @@ -171,7 +171,7 @@ def get_intratec_data() -> pd.DataFrame: def get_raw_technology_mapping( - module: Literal["energy", "materials"], + module: Literal["energy", "materials", "cooling"], ) -> pd.DataFrame: """Retrieve a technology mapping for `module`. @@ -234,7 +234,7 @@ def subset_module_map(raw_map): def adjust_technology_mapping( - module: Literal["energy", "materials"], + module: Literal["energy", "materials", "cooling"], ) -> pd.DataFrame: """Adjust technology mapping based on sources and assumptions.