fix mypy

modyn/trainer_server/internal/trainer/maybe_measure_gpu_ops.py

@@ -6,14 +6,15 @@
 
 
 class MaybeMeasureGPUOps:
+    # type: ignore[no-untyped-def]
     def __init__(self, measure: bool, measurement_name: str, device: str, stop_watch: Stopwatch, **kwargs) -> None:
         self._device = device
         self._measure = measure
         self._measurement_name = measurement_name
         self._kwargs = kwargs
         self._stop_watch = stop_watch
 
-    def __enter__(self):
+    def __enter__(self) -> None:
         if self._measure:
             torch.cuda.synchronize(device=self._device)
             self._stop_watch.start(name=self._measurement_name, **self._kwargs)