use pytorch-ie from ArneBinder/pytorch-ie#363

ArneBinder · Nov 7, 2023 · 053a2e7 · 053a2e7
1 parent 06c6b0e
commit 053a2e7
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 12 deletions.
diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -24,7 +24,8 @@ classifiers = [
 
 [tool.poetry.dependencies]
 python = "^3.9"
-pytorch-ie = ">=0.26.0,<0.27.0"
+#pytorch-ie = ">=0.26.0,<0.27.0"
+pytorch-ie = { git = "https://github.com/ChristophAlt/pytorch-ie.git", branch = "decouple_pie_dataset" }
 
 [tool.poetry.group.dev.dependencies]
 torch = {version = "^2.1.0+cpu", source = "pytorch"}