From bcebbc08ee04fc8acaf05aef8bb5bff69a045ddd Mon Sep 17 00:00:00 2001
From: Arne Binder <arne.binder@dfki.de>
Date: Tue, 7 Nov 2023 00:46:42 +0100
Subject: [PATCH] use pytorch-ie 0.26.1.dev1699314147 (pre-release on test
 pypi)

---
 poetry.lock    | 29 +++++++++++++++--------------
 pyproject.toml |  2 +-
 2 files changed, 16 insertions(+), 15 deletions(-)

diff --git a/poetry.lock b/poetry.lock
index b9fcea87..46f94b0d 100644
--- a/poetry.lock
+++ b/poetry.lock
@@ -1217,27 +1217,28 @@ six = ">=1.5"
 
 [[package]]
 name = "pytorch-ie"
-version = "0.26.0"
+version = "0.26.1.dev1699314147"
 description = "State-of-the-art Information Extraction in PyTorch"
 optional = false
-python-versions = "^3.9"
-files = []
-develop = false
+python-versions = ">=3.9,<4.0"
+files = [
+    {file = "pytorch_ie-0.26.1.dev1699314147-py3-none-any.whl", hash = "sha256:10e8aa445058b29cf4ae2a8e905f5614f803afb29ebad93f75e1f4fb1fde32bd"},
+    {file = "pytorch_ie-0.26.1.dev1699314147.tar.gz", hash = "sha256:01a398eef1874b5de464941774bbbeeef0ef862b5435a4523ade110f7707934e"},
+]
 
 [package.dependencies]
-absl-py = "^1.0.0"
-datasets = "^2.13"
+absl-py = ">=1.0.0,<2.0.0"
+datasets = ">=2.13,<3.0"
 fsspec = "<2023.9.0"
-pytorch-lightning = "^2"
+pytorch-lightning = ">=2,<3"
 torch = ">=1.10"
-torchmetrics = "^1"
-transformers = "^4.18"
+torchmetrics = ">=1,<2"
+transformers = ">=4.18,<5.0"
 
 [package.source]
-type = "git"
-url = "https://github.com/ChristophAlt/pytorch-ie.git"
-reference = "decouple_pie_dataset"
-resolved_reference = "15862d54de0066d2dee0f69fd7bf27527bbfb81d"
+type = "legacy"
+url = "https://test.pypi.org/simple"
+reference = "pre-release"
 
 [[package]]
 name = "pytorch-lightning"
@@ -2161,4 +2162,4 @@ multidict = ">=4.0"
 [metadata]
 lock-version = "2.0"
 python-versions = "^3.9"
-content-hash = "405005251a71a59558ec8b03c8b5bc44e84682b9035fe6a1053f864dec2c7e54"
+content-hash = "8e15677d9874e0189266d03d9fb22cf968f9b1437476fd41ab0010d2551a6975"
diff --git a/pyproject.toml b/pyproject.toml
index d24789e7..2f483f92 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -25,7 +25,7 @@ classifiers = [
 [tool.poetry.dependencies]
 python = "^3.9"
 #pytorch-ie = ">=0.26.0,<0.27.0"
-pytorch-ie = { git = "https://github.com/ChristophAlt/pytorch-ie.git", branch = "decouple_pie_dataset" }
+pytorch-ie = { version = "0.26.1.dev1699314147", source = "pre-release" }
 
 [tool.poetry.group.dev.dependencies]
 torch = {version = "^2.1.0+cpu", source = "pytorch"}