Merge branch 'develop' into dashes-in-hostnames

ls1intum · Oct 22, 2023 · 75b8177 · 75b8177
2 parents 9f41859 + 81b677a
commit 75b8177
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/module_programming_themisml/Dockerfile b/module_programming_themisml/Dockerfile
@@ -24,9 +24,7 @@ RUN poetry config virtualenvs.create true \
     && poetry install --no-interaction --no-ansi
 
 # Re-install pytorch in the CPU version because we don't have a GPU
-RUN poetry source add --priority=supplemental torch-stable https://download.pytorch.org/whl/
-# Install PyTorch:
-RUN poetry add torch==2.0.0+cpu
+RUN poetry add https://download.pytorch.org/whl/cpu/torch-2.1.0%2Bcpu-cp310-cp310-linux_x86_64.whl
 # Somehow, the torch library needs sympy to be installed manually as well
 RUN poetry add sympy==1.12
 

diff --git a/module_programming_themisml/pyproject.toml b/module_programming_themisml/pyproject.toml
@@ -10,6 +10,7 @@ python = "^3.10"
 athena = {path = "../athena", develop = true}
 antlr4-python3-runtime = "^4.13.1"
 code-bert-score = "^0.4.1"
+torch = "^2.1.0"
 
 [tool.poetry.scripts]
 module = "athena:run_module"