Merge pull request #2 from wassname/just_poetry

Just poetry
This commit is contained in:
Michael J Clark
2024-09-02 14:23:43 +08:00
committed by GitHub
+9 -4
View File
@@ -6,6 +6,9 @@ authors = ["{{ cookiecutter.author_name }}"]
license = "MIT"
readme = "README.md"
[virtualenvs]
in-project=true
[tool.poetry.dependencies]
python = ">=3.10"
numpy = "^1.26.1"
@@ -27,16 +30,18 @@ tqdm = "^4.66.1"
# loguru = "^0.7.2"
# einops = "^0.3.1"
[tool.poetry.group.dev.dependencies]
ipykernel = "^6.25.2"
ruff = "^0.1.3"
[[tool.poetry.source]]
# pytorch cuda needs to compe from another source https://python-poetry.org/docs/dependency-specification/#source-dependencies
name = "pytorch"
url = "https://download.pytorch.org/whl/cu124"
priority = "explicit"
[tool.poetry.group.dev.dependencies]
ipykernel = "^6.25.2"
ipywidgets = "^8.1.3"
ruff = "^0.1.3"
pylama = "^8.4.1"
[build-system]
requires = ["poetry-core"]
build-backend = "poetry.core.masonry.api"