From 3ede32d3f9af8a8707cad337a131b217d27c0aea Mon Sep 17 00:00:00 2001
From: Florimond Manca <florimond.manca@datadoghq.com>
Date: Mon, 16 Mar 2020 13:05:12 +0100
Subject: [PATCH] Relax type of `ServiceCheck` enum items

---
 datadog_checks_base/datadog_checks/base/types.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/datadog_checks_base/datadog_checks/base/types.py b/datadog_checks_base/datadog_checks/base/types.py
index 1354f46bba5ea..ed97bc44c204f 100644
--- a/datadog_checks_base/datadog_checks/base/types.py
+++ b/datadog_checks_base/datadog_checks/base/types.py
@@ -17,7 +17,12 @@
 ServiceCheckStatus = Literal[0, 1, 2, 3]  # Can serve as an int enum type for type checking purposes.
 _ServiceCheckType = NamedTuple(
     '_ServiceCheckType',
-    [('OK', Literal[0]), ('WARNING', Literal[1]), ('CRITICAL', Literal[2]), ('UNKNOWN', Literal[3])],
+    [
+        ('OK', ServiceCheckStatus),
+        ('WARNING', ServiceCheckStatus),
+        ('CRITICAL', ServiceCheckStatus),
+        ('UNKNOWN', ServiceCheckStatus),
+    ],
 )
 ServiceCheck = _ServiceCheckType(0, 1, 2, 3)  # For public enum-style use: `ServiceCheck.OK`, ...