From 19736487c72b69df93be478720630ac1d39c5623 Mon Sep 17 00:00:00 2001 From: Tamas Molnar Date: Tue, 3 Dec 2024 17:52:49 +0100 Subject: [PATCH] Fix python classification example (#487) Argument was mssing from classifier function --- README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/README.md b/README.md index 7721bb22..07fbebd3 100644 --- a/README.md +++ b/README.md @@ -447,7 +447,7 @@ engine_args = EngineArgs( engine="torch", model_warmup=True) array = AsyncEngineArray.from_args([engine_args]) -async def classifier(): +async def classifier(engine: AsyncEmbeddingEngine): async with engine: predictions, usage = await engine.classify(sentences=sentences) # or handle the async start / stop yourself.