Only delete keys if they exist, which depending on the invocation the…

…y may not.
rapidsai · Feb 15, 2022 · 05dc395 · 05dc395
1 parent 57608eb
commit 05dc395
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/python/cudf/cudf/tests/conftest.py b/python/cudf/cudf/tests/conftest.py
@@ -32,5 +32,8 @@ def pytest_sessionfinish(session, exitstatus):
     Called after whole test run finished, right before
     returning the exit status to the system.
     """
-    del os.environ["NO_EXTERNAL_ONLY_APIS"]
-    del os.environ["_CUDF_TEST_ROOT"]
+    try:
+        del os.environ["NO_EXTERNAL_ONLY_APIS"]
+        del os.environ["_CUDF_TEST_ROOT"]
+    except KeyError:
+        pass