From 542bc733d6ebb6da2554704fc101830a07791584 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Fabian=20Prei=C3=9F?= Date: Mon, 10 Jun 2024 21:13:50 +0200 Subject: [PATCH] Fix missing numpy dependency in pyproject.toml (#524) --- python/pyproject.toml | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/python/pyproject.toml b/python/pyproject.toml index a932a263e..95ec66102 100644 --- a/python/pyproject.toml +++ b/python/pyproject.toml @@ -16,13 +16,14 @@ classifiers = [ dependencies = [ "requests", "tqdm", + "numpy", ] [project.optional-dependencies] srt = ["aiohttp", "fastapi", "psutil", "rpyc", "torch", "uvloop", "uvicorn", "zmq", "vllm==0.4.3", "interegular", "pydantic", "pillow", "packaging", "huggingface_hub", "hf_transfer", "outlines>=0.0.34"] -openai = ["openai>=1.0", "numpy", "tiktoken"] -anthropic = ["anthropic>=0.20.0", "numpy"] +openai = ["openai>=1.0", "tiktoken"] +anthropic = ["anthropic>=0.20.0"] litellm = ["litellm>=1.0.0"] all = ["sglang[srt]", "sglang[openai]", "sglang[anthropic]", "sglang[litellm]"]