diff --git a/src/common/method_wrappers/Decorator.py b/src/common/method_wrappers/Decorator.py index 197378dadc2ff85c673c3b4929c2cc6bfd6a3383..98ade3fc0d382579e330d1fa8b8497d4c82471e6 100644 --- a/src/common/method_wrappers/Decorator.py +++ b/src/common/method_wrappers/Decorator.py @@ -76,7 +76,7 @@ class MetricsPool: def get_metrics( self, method : str, labels : Optional[Dict[str, str]] = None - ) -> Tuple[Histogram, Counter, Counter, Counter]: + ) -> Tuple[MetricWrapperBase, MetricWrapperBase, MetricWrapperBase, MetricWrapperBase]: histogram_duration : Histogram = self.get_or_create(method, MetricTypeEnum.HISTOGRAM_DURATION) counter_started : Counter = self.get_or_create(method, MetricTypeEnum.COUNTER_STARTED) counter_completed : Counter = self.get_or_create(method, MetricTypeEnum.COUNTER_COMPLETED)