explicit connectionError import

DataDog · Jul 29, 2021 · 9f365c3 · 9f365c3
1 parent 8c1d6a8
commit 9f365c3
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/ibm_db2/datadog_checks/ibm_db2/ibm_db2.py b/ibm_db2/datadog_checks/ibm_db2/ibm_db2.py
@@ -5,6 +5,7 @@
 
 from itertools import chain
 from time import time as timestamp
+from requests import ConnectionError
 
 import ibm_db
 

diff --git a/ibm_db2/tests/test_unit.py b/ibm_db2/tests/test_unit.py
@@ -4,6 +4,8 @@
 import mock
 import pytest
 
+from requests import ConnectionError
+
 from datadog_checks.ibm_db2 import IbmDb2Check
 from datadog_checks.ibm_db2.utils import scrub_connection_string