diff --git a/pyproject.toml b/pyproject.toml index a426a420..673be85a 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -19,8 +19,8 @@ classifiers = [ license = {file = "LICENSE"} dependencies = [ # Core ML/AI packages - "torch>=2.6", - "transformers>=4.5", + "torch", + "transformers", # Data processing "datasets", @@ -50,6 +50,7 @@ dependencies = [ "pydantic", "wrapt", "asgiref>=3.7.0", # Async/sync conversion utilities + "wandb", # Development and testing "pytest", @@ -63,14 +64,27 @@ dependencies = [ "mkdocstrings[python]>=0.24.0", "mkdocs-autorefs>=0.5.0", "pymdown-extensions>=10.0.0", +] + +[project.optional-dependencies] +verl = [ + "verl==0.6.1", + "torch>=2.8.0", + "torchvision>=0.23.0", + "vllm>=0.10.2,<=0.11.0", + "flash-attn>=2.8.3", + "qwen-vl-utils", +] - # Tinker +tinker = [ "tinker", "tinker-cookbook @ git+https://github.com/thinking-machines-lab/tinker-cookbook.git#egg=tinker-cookbook", ] -[project.optional-dependencies] +sglang = [ + "sglang>=0.5.5", +] smolagents = [ "smolagents==1.20.0", @@ -80,7 +94,6 @@ strands = [ "strands-agents", ] - swe = [ "docker", "kubernetes",