Add cooling variant to type hints

iiasa · Oct 22, 2024 · 09e4a65 · 09e4a65
1 parent be7bea6
commit 09e4a65
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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.