handle disabled flag

coiled · Oct 23, 2024 · eea37f2 · eea37f2
1 parent b05da89
commit eea37f2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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: