fix(tests): do not reinstall from main in doc test

IML-DKFZ · Apr 25, 2023 · 45d3a64 · 45d3a64
1 parent 16aa3a0
commit 45d3a64
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/docs/extending_fd-shifts.ipynb b/docs/extending_fd-shifts.ipynb
@@ -21,7 +21,11 @@
    "cell_type": "code",
    "execution_count": null,
    "id": "4e4a7b52-102b-4bca-b51a-782dd99fe58b",
-   "metadata": {},
+   "metadata": {
+    "tags": [
+     "skip-execution"
+    ]
+   },
    "outputs": [],
    "source": [
     "!wget https://github.com/korakot/kora/releases/download/v0.10/py310.sh\n",

diff --git a/pyproject.toml b/pyproject.toml
@@ -33,7 +33,7 @@ dependencies = [
   "tensorboard>=2.4.1",
   "timm==0.5.4",
   "toml>=0.10.2",
-  "torch==1.11.0",
+  "torch>=1.11.0,<2.0.0",
   "torchmetrics>=0.2.0",
   "torchvision>=0.12.0",
   "tqdm>=4.62.0",