typing

Author: djdameln

src/anomalib/metrics/base.py

@@ -178,7 +178,7 @@ def update(self, batch: Batch, *args, **kwargs) -> None:
         values = [getattr(batch, key) for key in self.fields]
         super().update(*values, *args, **kwargs)  # type: ignore[misc]
 
-    def compute(self) -> torch.Tensor:
+    def compute(self) -> torch.Tensor | None:
         """Compute the metric value.
 
         If the metric has not been updated, and metric is not in strict mode, return None.