Add cooling variant to type hints

iiasa · Sep 11, 2024 · c445299 · c445299
1 parent 8611cbd
commit c445299
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.