Relax type constraints Timestamp

csmarchbanks · csmarchbanks · commit 557d123b349f · 2022-01-28T08:30:01.000-07:00
Timestamps in samples or exemplars can be floats or Timestamp objects.

Signed-off-by: Chris Marchbanks &lt;csmarchbanks@gmail.com&gt;
diff --git a/prometheus_client/samples.py b/prometheus_client/samples.py
@@ -1,4 +1,4 @@
-from typing import Dict, NamedTuple, Optional
+from typing import Dict, NamedTuple, Optional, Union
 
 
 class Timestamp:
@@ -39,12 +39,12 @@ def __gt__(self, other):
 class Exemplar(NamedTuple):
     labels: Dict[str, str]
     value: float
-    timestamp: Optional[float] = None
+    timestamp: Optional[Union[float, Timestamp]] = None
 
 
 class Sample(NamedTuple):
     name: str
     labels: Dict[str, str]
     value: float
-    timestamp: Optional[float] = None
+    timestamp: Optional[Union[float, Timestamp]] = None
     exemplar: Optional[Exemplar] = None
