add typehint for dict

funkelab · Aug 29, 2024 · b438fc0 · b438fc0
1 parent bc5e6b0
commit b438fc0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/gunpowder/torch/nodes/train.py b/gunpowder/torch/nodes/train.py
@@ -123,7 +123,7 @@ def __init__(
         # not yet implemented
         gradients = gradients
         loss_inputs = {f"loss_{k}": v for k, v in loss_inputs.items()}
-        all_inputs = {f"{k}": v for k, v in inputs.items() if v not in outputs.values()}
+        all_inputs: dict[str | int, Any] = {f"{k}": v for k, v in inputs.items() if v not in outputs.values()}
         all_inputs.update(
             {k: v for k, v in loss_inputs.items() if v not in outputs.values()}
         )