Add type annotation for max_processes_sample

con · Jan 14, 2025 · 02ef9b1 · 02ef9b1
1 parent 5040f1d
commit 02ef9b1
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/test/test_e2e.py b/test/test_e2e.py
@@ -2,6 +2,7 @@
 import json
 from pathlib import Path
 import subprocess
+from typing import Dict
 import pytest
 
 ABANDONING_PARENT = str(Path(__file__).with_name("data") / "abandoning_parent.sh")
@@ -21,7 +22,7 @@ def test_abandoning_parent(temp_output_dir: str, num_children: int) -> None:
     with open(f"{duct_prefix}usage.json") as usage_file:
         all_samples = [json.loads(line) for line in usage_file]
 
-    max_processes_sample = {"processes": {}}
+    max_processes_sample: Dict[str, Dict] = {"processes": {}}
     for sample in all_samples:
         if len(max_processes_sample) < len(sample.get("processes")):
             max_processes_sample = sample