Declare registry on MetricWrapperBase as Optional (prometheus#754)

reivilibre · web-flow · commit b44b63e59b16 · 2022-01-28T08:24:21.000-07:00
* Declare `registry` on `MetricWrapperBase` as `Optional`
* Declare `registry` on `Enum` as `Optional`

Signed-off-by: Olivier Wilkinson (reivilibre) &lt;oliverw@matrix.org&gt;
diff --git a/prometheus_client/metrics.py b/prometheus_client/metrics.py
@@ -107,7 +107,7 @@ def __init__(self: T,
                  namespace: str = '',
                  subsystem: str = '',
                  unit: str = '',
-                 registry: CollectorRegistry = REGISTRY,
+                 registry: Optional[CollectorRegistry] = REGISTRY,
                  _labelvalues: Optional[Sequence[str]] = None,
                  ) -> None:
         self._name = _build_full_name(self._type, name, namespace, subsystem, unit)
@@ -677,7 +677,7 @@ def __init__(self,
                  namespace: str = '',
                  subsystem: str = '',
                  unit: str = '',
-                 registry: CollectorRegistry = REGISTRY,
+                 registry: Optional[CollectorRegistry] = REGISTRY,
                  _labelvalues: Optional[Sequence[str]] = None,
                  states: Optional[Sequence[str]] = None,
                  ):
