diff options
-rw-r--r-- | tests/testutils/artifactshare.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tests/testutils/artifactshare.py b/tests/testutils/artifactshare.py index e5434b63f..18ecc5e3e 100644 --- a/tests/testutils/artifactshare.py +++ b/tests/testutils/artifactshare.py @@ -74,11 +74,11 @@ class ArtifactShare(): signal.signal(signal.SIGTERM, lambda signalnum, frame: sys.exit(0)) try: - import pytest_cov + from pytest_cov.embed import cleanup_on_sigterm except ImportError: pass else: - pytest_cov.embed.cleanup_on_sigterm() + cleanup_on_sigterm() server = stack.enter_context( create_server( |