[tool.poetry]
name = "icd10_docker"
version = "0.1.0"
description = ""
authors = ["lyangasovid <lyangasovid@gmail.com>"]

[tool.poetry.dependencies]
python = "^3.9"
scikit-learn = "1.3.0"
transformers = "4.29.2"
flask = "2.0.3"
xgboost = "1.7.6"
werkzeug = "2.2.2"
numpy = "1.26.3"
pandas = "2.2.0"
tqdm = "4.66.1"
torch = {version = "^2.3.1+cpu", source = "pytorch_cpu"}
torchvision = {version = "^0.18.1+cpu", source = "pytorch_cpu"}

[tool.poetry.dev-dependencies]

[[tool.poetry.source]]
name = "pytorch_cpu"
url = "https://download.pytorch.org/whl/cpu"
priority = "explicit"

[build-system]
requires = ["poetry-core>=1.0.0"]
build-backend = "poetry.core.masonry.api"