Add type hints for technology sets

iiasa · Dec 3, 2024 · 5993c55 · 5993c55
1 parent 8ed0523
commit 5993c55
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/message_ix_models/tests/tools/costs/test_decay.py b/message_ix_models/tests/tools/costs/test_decay.py
@@ -26,7 +26,7 @@
     ),
 )
 def test_get_module_scenarios_reduction(
-    module: Literal["energy", "materials", "cooling"], t_exp
+    module: Literal["energy", "materials", "cooling"], t_exp: set[str]
 ) -> None:
     tech_map = energy_map = get_raw_technology_mapping("energy")
 
@@ -57,7 +57,7 @@ def test_get_module_scenarios_reduction(
     ),
 )
 def test_get_module_cost_reduction(
-    module: Literal["energy", "materials", "cooling"], t_exp
+    module: Literal["energy", "materials", "cooling"], t_exp: set[str]
 ) -> None:
     tech_map = energy_map = get_raw_technology_mapping("energy")
 
@@ -111,7 +111,9 @@ def test_get_technology_reduction_scenarios_data(
     ),
 )
 def test_project_ref_region_inv_costs_using_reduction_rates(
-    module: Literal["energy", "materials", "cooling"], t_exp, t_excluded
+    module: Literal["energy", "materials", "cooling"],
+    t_exp: set[str],
+    t_excluded: set[str],
 ) -> None:
     # Set up
     config = Config(module=module)