Ignore variable as key in TypedDict

vrplib/parse/parse_solution.py

@@ -1,8 +1,11 @@
-from typing import Dict, List, Union
+from typing import List, TypedDict
 
 from .parse_utils import infer_type, text2lines
 
-Solution = Dict[str, Union[float, str, List]]
+
+class Solution(TypedDict, total=False):
+    routes: List[List[int]]
+    cost: float
 
 
 def parse_solution(text: str) -> Solution:
@@ -25,11 +28,11 @@ def parse_solution(text: str) -> Solution:
     for line in text2lines(text):
         if "Route" in line:
             route = [int(idx) for idx in line.split(":")[1].split(" ") if idx]
-            solution["routes"].append(route)  # type: ignore
+            solution["routes"].append(route)
         elif ":" in line or " " in line:  # Split at first colon or whitespace
             split_at = ":" if ":" in line else " "
             k, v = [word.strip() for word in line.split(split_at, 1)]
-            solution[k.lower()] = infer_type(v)
+            solution[k] = infer_type(v)  # type: ignore
         else:  # Ignore lines without keyword-value pairs
             continue
 

vrplib/parse/parse_vrplib.py

@@ -38,11 +38,11 @@ def parse_vrplib(text: str, compute_edge_weights: bool = True) -> Instance:
 
     for spec in specs:
         key, value = parse_specification(spec)
-        instance[key] = value
+        instance[key] = value  # type: ignore
 
     for section in sections:
         section, data = parse_section(section, instance)
-        instance[section] = data
+        instance[section] = data  # type: ignore
 
     if instance and compute_edge_weights and "edge_weight" not in instance:
         # Compute edge weights if there was no explicit edge weight section