diff --git a/tests/conftest.py b/tests/conftest.py index 0f8652484e..d1f64da2c3 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -884,7 +884,7 @@ def memray_profile( memray_option = pytestconfig.getoption("--memray") if memray_option == "none": - yield + yield contextlib.nullcontext elif memray_option != "scheduler": raise ValueError(f"Unhandled value for --memray: {memray_option}") else: