Fix type hint

Markus28 · Markus28 · commit dbefab11efc9 · 2025-08-06T14:07:27.000+02:00
diff --git a/polygraph/metrics/base/mmd.py b/polygraph/metrics/base/mmd.py
@@ -197,11 +197,7 @@ def _generate_mmd_samples(
     @abstractmethod
     def compute(
         *args, **kwargs
-    ) -> Union[
-        MetricInterval,
-        Dict[str, float],
-        Tuple[Union[MetricInterval, Dict[str, float]], np.ndarray],
-    ]: ...
+    ) -> MetricInterval: ...
 
 
 class DescriptorMMD2Interval(_DescriptorMMD2Interval, GenerationMetricInterval):
