summaryrefslogtreecommitdiff
path: root/pyproject.toml
diff options
context:
space:
mode:
Diffstat (limited to 'pyproject.toml')
-rw-r--r--pyproject.toml6
1 files changed, 4 insertions, 2 deletions
diff --git a/pyproject.toml b/pyproject.toml
index 49bb049..43943e7 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -24,7 +24,6 @@ h5py = "^2.10.0"
toml = "^0.10.1"
torch = "^1.6.0"
torchvision = "^0.7.0"
-torchsummary = "^1.5.1"
loguru = "^0.5.0"
matplotlib = "^3.2.1"
tqdm = "^4.46.1"
@@ -32,7 +31,10 @@ pytest = "^5.4.3"
opencv-python = "^4.3.0"
nltk = "^3.5"
einops = "^0.2.0"
-wandb = "^0.9.5"
+wandb = "^0.9.6"
+torch-summary = "^1.4.2"
+python-Levenshtein = "^0.12.0"
+defusedxml = "^0.6.0"
[tool.poetry.dev-dependencies]
pytest = "^5.4.2"