Fix conftest to treat harness as context manager

dominictarro · Apr 15, 2024 · 049ccf1 · 049ccf1
1 parent 307ed79
commit 049ccf1
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/tests/conftest.py b/tests/conftest.py
@@ -11,7 +11,5 @@
 @pytest.fixture
 def harness():
     """Return a `prefect_test_harness`."""
-    try:
-        prefect_test_harness()
-    finally:
-        pass
+    with prefect_test_harness():
+        yield