summaryrefslogtreecommitdiff
path: root/pyproject.toml
diff options
context:
space:
mode:
authoraktersnurra <gustaf.rydholm@gmail.com>2020-09-20 01:39:22 +0200
committeraktersnurra <gustaf.rydholm@gmail.com>2020-09-20 01:39:22 +0200
commit6137f43c910946301279825e50759a9dd76c6131 (patch)
treefaa11c77de6602b17629b7bab1081264907a1925 /pyproject.toml
parentdca26eff11428ef43caef534c29f0b54faa3ff7b (diff)
fixes.
Diffstat (limited to 'pyproject.toml')
-rw-r--r--pyproject.toml4
1 files changed, 4 insertions, 0 deletions
diff --git a/pyproject.toml b/pyproject.toml
index 43943e7..43ab115 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -35,6 +35,10 @@ wandb = "^0.9.6"
torch-summary = "^1.4.2"
python-Levenshtein = "^0.12.0"
defusedxml = "^0.6.0"
+pytorch-block-sparse = "^0.1.2"
+pytorch-metric-learning = "^0.9.92"
+faiss-gpu = "^1.6.3"
+omegaconf = "^2.0.2"
[tool.poetry.dev-dependencies]
pytest = "^5.4.2"