wassname/cookiecutter-data-science
1
0
Fork 0
mirror of https://github.com/wassname/cookiecutter-data-science.git synced 2026-06-27 16:30:02 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update pyproject.toml

Browse Source
This commit is contained in:
Michael J Clark
2024-05-10 15:03:18 +08:00
committed by GitHub
parent 63dd6f602d
commit 10b39d0e7d
1 changed files with 13 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files
{{ cookiecutter.repo_name }}/pyproject.toml
+13 -11
View File
@@ -8,18 +8,18 @@ readme = "README.md"
[tool.poetry.dependencies]
python = ">=3.10,<3.13"
torch = {version = "^2.1.0+cu124", source = "pytorch"}
simple-parsing = "^0.1.4"
# simple-parsing = "^0.1.4"
tqdm = "^4.66.1"
numpy = "^1.26.1"
pandas = "^2.1.1"
lightning = "^2.1.0"
matplotlib = "^3.8.0"
loguru = "^0.7.2"
einops = "^0.7.0"
scikit-learn = "^1.3.1"
pytorch-optimizer = "^2.12.0"
torchinfo = "^1.8.0"
# torch = {version = "^2.1.0+cu124", source = "pytorch"}
# lightning = "^2.1.0"
# pytorch-optimizer = "^2.12.0"
# torchinfo = "^1.8.0"
# accelerate = "^0.24.1"
# transformers = "4.34.0"
# accelerate = "^0.24.1"
@@ -27,17 +27,19 @@ torchinfo = "^1.8.0"
# loguru = "^0.7.2"
# einops = "^0.3.1"
[tool.poetry.group.dev.dependencies]
ipykernel = "^6.25.2"
ruff = "^0.1.3"
[[tool.poetry.source]]
name = "pytorch"
url = "https://download.pytorch.org/whl/cu124"
priority = "explicit"
[tool.poetry.group.dev.dependencies]
ipykernel = "^6.25.2"
ruff = "^0.1.3"
pylama = "^8.4.1"
[build-system]
requires = ["poetry-core"]
build-backend = "poetry.core.masonry.api"
[virtualenvs]
create = true
in-project = true