From f7b1c3320876dea4d10b6e915e53e6c568175119 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Fri, 19 Jul 2024 16:40:54 -0700 Subject: [PATCH] fix typing errors --- litellm/batches/main.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/litellm/batches/main.py b/litellm/batches/main.py index db220e15faa..79aefa5f518 100644 --- a/litellm/batches/main.py +++ b/litellm/batches/main.py @@ -50,7 +50,7 @@ async def acreate_batch( extra_headers: Optional[Dict[str, str]] = None, extra_body: Optional[Dict[str, str]] = None, **kwargs, -) -> Coroutine[Any, Any, Batch]: +) -> Batch: """ Async: Creates and executes a batch from an uploaded file of request @@ -189,7 +189,7 @@ async def aretrieve_batch( extra_headers: Optional[Dict[str, str]] = None, extra_body: Optional[Dict[str, str]] = None, **kwargs, -) -> Coroutine[Any, Any, Batch]: +) -> Batch: """ Async: Retrieves a batch.