Do not require pytest_timeout (#6224)

dask · Apr 27, 2022 · b837003 · b837003
1 parent b6a62f8
commit b837003
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/distributed/utils_test.py b/distributed/utils_test.py
@@ -77,7 +77,15 @@
 except ImportError:
     pass
 
-from pytest_timeout import is_debugging
+try:
+    from pytest_timeout import is_debugging
+except ImportError:
+
+    def is_debugging() -> bool:
+        # The pytest_timeout logic is more sophisticated. Not only debuggers
+        # attach a trace callback but vendoring the entire logic is not worth it
+        return sys.gettrace() is not None
+
 
 logger = logging.getLogger(__name__)