Fix review comments

Pliner · Pliner · commit 32535efdafd3 · 2022-02-01T14:14:01.000+05:00
diff --git a/prometheus_client/metrics.py b/prometheus_client/metrics.py
@@ -2,7 +2,7 @@
 import time
 import types
 from typing import (
-    Any, Callable, Dict, Iterable, Optional, Sequence, Type, TypeVar,
+    Any, Callable, Dict, Iterable, Optional, Sequence, Type, TypeVar, Union,
 )
 
 from . import values  # retain this import style for testability
@@ -538,7 +538,7 @@ def __init__(self,
                  unit: str = '',
                  registry: Optional[CollectorRegistry] = REGISTRY,
                  _labelvalues: Optional[Sequence[str]] = None,
-                 buckets: Sequence[float] = DEFAULT_BUCKETS,
+                 buckets: Sequence[Union[float, int, str]] = DEFAULT_BUCKETS,
                  ):
         self._prepare_buckets(buckets)
         super().__init__(
@@ -553,7 +553,7 @@ def __init__(self,
         )
         self._kwargs['buckets'] = buckets
 
-    def _prepare_buckets(self, buckets: Sequence[float]) -> None:
+    def _prepare_buckets(self, buckets: Sequence[Union[float, int, str]]) -> None:
         buckets = [float(b) for b in buckets]
         if buckets != sorted(buckets):
             # This is probably an error on the part of the user,
