fix mypy error

Signed-off-by: Soumya Snigdha Kundu <soumya_snigdha.kundu@kcl.ac.uk>

Author: aymuos15

monai/engines/utils.py

@@ -423,8 +423,11 @@ def __call__(self, engine: Any, batchdata: dict[str, Any]) -> dict:
         """
         acc = self.accumulation_steps
 
+        result: dict
+
         if acc == 1:
-            return engine._iteration(engine, batchdata)
+            result = engine._iteration(engine, batchdata)
+            return result
 
         # engine.state.iteration is 1-indexed and already incremented before __call__
         epoch_length = engine.state.epoch_length  # None for iterable datasets

tests/engines/test_gradient_accumulation.py

@@ -12,6 +12,7 @@
 from __future__ import annotations
 
 import unittest
+from typing import Any
 from unittest.mock import MagicMock
 
 import torch
@@ -201,7 +202,7 @@ def test_scaler_is_none(self) -> None:
 
     def test_batch_data_passed_correctly(self) -> None:
         engine = _make_engine(epoch_length=4, iteration=1)
-        test_batch = {CommonKeys.IMAGE: torch.randn(1, 10), CommonKeys.LABEL: torch.randn(1, 1)}
+        test_batch: dict[str, Any] = {CommonKeys.IMAGE: torch.randn(1, 10), CommonKeys.LABEL: torch.randn(1, 1)}
 
         GradientAccumulation(accumulation_steps=2)(engine, test_batch)