verifiers 0.2.2.dev27__tar.gz → 0.2.2.dev29__tar.gz

{verifiers-0.2.2.dev27 → verifiers-0.2.2.dev29}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: verifiers
-Version: 0.2.2.dev27
+Version: 0.2.2.dev29
 Summary: Verifiers: Environments for LLM Reinforcement Learning
 Project-URL: Homepage, https://github.com/primeintellect-ai/verifiers
 Project-URL: Documentation, https://github.com/primeintellect-ai/verifiers

{verifiers-0.2.2.dev27 → verifiers-0.2.2.dev29}/tests/test_math_rubric.py

@@ -51,9 +51,11 @@ class TestMathRubric:
         state["trajectory"] = []
         state["timing"] = RolloutTiming()
 
-        await rubric.score_rollout(state)
-
-        assert state["metrics"]["correct_answer"] == 1.0
+        try:
+            await rubric.score_rollout(state)
+            assert state["metrics"]["correct_answer"] == 1.0
+        finally:
+            await rubric.teardown()
 
     @pytest.mark.asyncio
     @pytest.mark.parametrize(
@@ -79,9 +81,11 @@ class TestMathRubric:
         state["trajectory"] = []
         state["timing"] = RolloutTiming()
 
-        await rubric.score_rollout(state)
-
-        assert state["metrics"]["correct_answer"] == 0.0
+        try:
+            await rubric.score_rollout(state)
+            assert state["metrics"]["correct_answer"] == 0.0
+        finally:
+            await rubric.teardown()
 
     @pytest.mark.asyncio
     @pytest.mark.parametrize("timeout_seconds", [0.1, 10])
@@ -106,13 +110,16 @@ class TestMathRubric:
         state["trajectory"] = []
         state["timing"] = RolloutTiming()
 
-        await rubric.score_rollout(state)
+        try:
+            await rubric.score_rollout(state)
 
-        # rollout should only pass for large timeout
-        if timeout_seconds == 10:
-            assert state["metrics"]["correct_answer"] == 1.0
-        else:
-            assert state["metrics"]["correct_answer"] == 0.0
+            # rollout should only pass for large timeout
+            if timeout_seconds == 10:
+                assert state["metrics"]["correct_answer"] == 1.0
+            else:
+                assert state["metrics"]["correct_answer"] == 0.0
+        finally:
+            await rubric.teardown()
 
 
 class TestVerifyResponseExceptionHandling: