remove nvidia from poetry lock

openclimatefix · Feb 27, 2024 · 9bec505 · 9bec505
1 parent ca489f2
commit 9bec505
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 153 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -19,6 +19,7 @@ ENV PIP_DEFAULT_TIMEOUT=100 \
 RUN pip install "poetry==$POETRY_VERSION"
 
 COPY pyproject.toml poetry.lock README.md .
+RUN pip install torch torchvision --index-url https://download.pytorch.org/whl/cpu
 RUN poetry install --only main --no-root
 
 COPY india_forecast_app ./india_forecast_app

diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -13,8 +13,9 @@ pandas = "1.5.3"
 pvnet = "3.0.5"
 pytz = "^2024.1"
 numpy = "^1.26.4"
-ocf-datapipes = "^3.2.9"
 huggingface-hub = "0.20.3"
+torch = "2.2.1"
+ocf-datapipes = "^3.2.9"
 
 [tool.poetry.group.dev.dependencies]
 pytest = "^7.4.4"