Better hack for INTERNAL ERROR (#217)

hauntsaninja · web-flow · commit a8002092a4e8 · 2025-09-27T01:12:21.000-07:00
diff --git a/mypy_primer/model.py b/mypy_primer/model.py
@@ -548,17 +548,21 @@ def canonicalise(line: str) -> str:
         net_change: dict[str, int] = defaultdict(int)
         for line in diff_lines:
             cline = canonicalise(line)
-            net_change[cline] += 1 if line[0] == "+" else -1
+            net_change[cline] += {"+": 1, "-": -1}.get(line[0], 0)
 
         output_lines: list[str] = []
         for line in diff_lines:
             cline = canonicalise(line)
-            if line[0] == "+" and net_change[cline] > 0:
-                output_lines.append(line)
-                net_change[cline] -= 1
-            elif line[0] == "-" and net_change[cline] < 0:
+            if line[0] == "+":
+                if net_change[cline] > 0:
+                    output_lines.append(line)
+                    net_change[cline] -= 1
+            elif line[0] == "-":
+                if net_change[cline] < 0:
+                    output_lines.append(line)
+                    net_change[cline] += 1
+            else:
                 output_lines.append(line)
-                net_change[cline] += 1
 
         return "\n".join(output_lines)
 
