Skip to content
This repository has been archived by the owner on Sep 24, 2024. It is now read-only.

Commit

Permalink
Updated pinned libs in pyproject.toml
Browse files Browse the repository at this point in the history
  • Loading branch information
aittalam committed Aug 6, 2024
1 parent 79c8e98 commit bae01c1
Showing 1 changed file with 22 additions and 24 deletions.
46 changes: 22 additions & 24 deletions pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -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"
Expand All @@ -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",
Expand Down

0 comments on commit bae01c1

Please sign in to comment.