From 8825033de7ce1851dbd9b19746508c3c2d4f7287 Mon Sep 17 00:00:00 2001 From: Alexandre Piche Date: Tue, 7 Jan 2025 18:15:01 +0000 Subject: [PATCH] clean up --- tapeagents/llms.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/tapeagents/llms.py b/tapeagents/llms.py index 5bfb3517..84b52f46 100644 --- a/tapeagents/llms.py +++ b/tapeagents/llms.py @@ -680,8 +680,7 @@ def batch_generate(self, prompts: list[Prompt]) -> list[LLMCall]: stream=self.stream, verify=False, ) - time_send_request = time.time() - start_send_request - self._stats["time_send_request"].append(time_send_request) + self._stats["time_send_request"].append(time.time() - start_send_request) if not r.ok: logger.error(f"Failed to get completion: {r.text}") r.raise_for_status()