diff --git a/pyproject.toml b/pyproject.toml index fb5bac0..02b26d1 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -26,47 +26,45 @@ dependencies = [ "wandb>=0.16.3", "protobuf>=3.20.2", "urllib3>=1.26.18,<2", - "pydantic>=2.6.4", - "pydantic-yaml>=1.2.0", "pydantic==2.6.4", "pydantic-yaml==1.2.0", "ray[default]==2.30.0", "loguru==0.7.2", - "s3fs", + "s3fs==2024.5.0", ] [project.optional-dependencies] -ruff = ["ruff==0.2.1"] +ruff = ["ruff==0.5.6"] jobs = [ # HuggingFace / pytorch - "torch==2.3.1", - "datasets>=2.17.1", - "transformers==4.38.0", - "accelerate==0.26.1", - "peft==0.7.1", - "trl==0.7.10", + "torch==2.4.0", + "datasets>=2.20.0", + "transformers==4.43.4", + "accelerate==0.33.0", + "peft==0.12.0", + "trl==0.9.6", "bitsandbytes==0.42.0", "bert_score==0.3.13", # Evaluation frameworks - "lm-eval==0.4.2", - "einops==0.7.0", - "openai==1.14.3", - "ragas==0.1.5", - "langchain-community==0.0.29", - "langchain_openai==0.1.1", + "lm-eval==0.4.3", + "einops==0.8.0", + "openai==1.38.0", + "ragas==0.1.12", + "langchain-community==0.2.11", + "langchain_openai==0.1.20", "sentencepiece==0.2.0", "evaluate==0.4.2", "mistralai==0.4.2", ] -test = ["pytest==7.4.3", "pytest-cov==4.1.0"] +test = ["pytest==8.3.2", "pytest-cov==5.0.0"] docs = [ - "Sphinx==7.2.6", - "nbsphinx==0.9.3", - "sphinx-codeautolink==0.15.0", - "myst-parser==2.0.0", - "furo==2024.1.29", + "Sphinx==7.4.7", + "nbsphinx==0.9.4", + "sphinx-codeautolink==0.15.2", + "myst-parser==3.0.1", + "furo==2024.7.18", ] -dev = ["lm-buddy[jobs,ruff,test,docs]", "pre-commit==3.6.0", "jupyter==1.0.0"] +dev = ["lm-buddy[jobs,ruff,test,docs]", "pre-commit==3.8.0", "jupyter==1.0.0"] [project.scripts] lm-buddy = "lm_buddy.cli:cli" @@ -82,7 +80,7 @@ addopts = "-v --cov src --no-cov-on-fail --disable-warnings" testpaths = ["tests"] [tool.ruff] -target-version = "py310" +target-version = "py311" exclude = [ ".bzr",