fix type errors

Author: henrykrumb

kvasircapsuleloader/metadata.py

@@ -68,7 +68,7 @@ def samples_by_class_by_patient(
         :return: _description_
         :rtype: Dict[FindingClass, Dict[str, List[KvasirCapsuleSample]]]
         """
-        S = {}
+        S: Dict[FindingClass, Dict[str, List[KvasirCapsuleSample]]] = {}
         for sample in self.samples:
             if sample.finding_class not in S:
                 S[sample.finding_class] = {}

kvasircapsuleloader/split.py

@@ -126,10 +126,10 @@ def save(self, path: Path):
         data = {
             "ratios": {**self._ratios},
             "seed": self._seed,
-            "samples": {key: [] for key in self._ratios},
+            "samples": {
+                phase: [sample.filename for sample in self.samples[phase]]
+                for phase in self._ratios
+            },
         }
-        for phase in self._ratios:
-            for sample in self.samples[phase]:
-                data["samples"][phase].append(sample.filename)
         with open(path, "w") as f:
             json.dump(data, f)

pyproject.toml

@@ -10,10 +10,13 @@ dependencies = [
     "click>=8.3.1",
     "mypy>=1.19.0",
     "pandas>=2.3.3",
+    "pandas-stubs>=2.3.3.251201",
     "pillow>=12.0.0",
     "pytest>=9.0.2",
     "requests>=2.32.5",
     "ruff>=0.14.8",
     "torch>=2.9.1",
     "tqdm>=4.67.1",
+    "types-requests>=2.32.4.20250913",
+    "types-tqdm>=4.67.0.20250809",
 ]