rename

Author: xzrderek

eval_protocol/pytest/evaluation_test_utils.py

@@ -465,8 +465,8 @@ async def execute_row_with_backoff_and_log(
 
     finally:
         # Prefer async cleanup if available, fall back to sync
-        if hasattr(rollout_processor, "aclose"):
-            await getattr(rollout_processor, "aclose")()
+        if hasattr(rollout_processor, "acleanup"):
+            await getattr(rollout_processor, "acleanup")()
         else:
             rollout_processor.cleanup()
 

eval_protocol/pytest/remote_rollout_processor.py

@@ -182,7 +182,7 @@ async def _sem_wrapper(r: EvaluationRow) -> EvaluationRow:
         tasks = [asyncio.create_task(_sem_wrapper(row)) for row in rows]
         return tasks
 
-    async def aclose(self) -> None:
+    async def acleanup(self) -> None:
         """Async cleanup - preferred when you can await."""
         if self._session and not self._session.closed:
             await self._session.close()
@@ -198,5 +198,5 @@ def cleanup(self) -> None:
                 # The session will be garbage collected eventually, but warn about it.
                 logger.warning(
                     "RemoteRolloutProcessor.cleanup() called outside of async context. "
-                    "Session may not be properly closed. Use `await processor.aclose()` when possible."
+                    "Session may not be properly closed. Use `await processor.acleanup()` when possible."
                 )

eval_protocol/training/gepa_trainer.py

@@ -504,8 +504,8 @@ async def evaluate_with_ep(
 
         finally:
             # Prefer async cleanup if available, fall back to sync
-            if hasattr(rollout_processor, "aclose"):
-                await getattr(rollout_processor, "aclose")()
+            if hasattr(rollout_processor, "acleanup"):
+                await getattr(rollout_processor, "acleanup")()
             else:
                 rollout_processor.cleanup()