fix type hint

Author: awaelchli

pytorch_lightning/loops/epoch/evaluation_epoch_loop.py

@@ -241,7 +241,7 @@ def _build_kwargs(self, batch: Any, batch_idx: int, dataloader_idx: int) -> Dict
 
     def _track_output_for_epoch_end(
         self, outputs: List[STEP_OUTPUT], output: Optional[STEP_OUTPUT]
-    ) -> List[Union[ResultCollection, Dict, Tensor]]:
+    ) -> List[STEP_OUTPUT]:
         if output is not None:
             if isinstance(output, dict):
                 output = recursive_detach(output, to_cpu=self.trainer.move_metrics_to_cpu)