fix: install pytorch-lightning in docker (#42)

Signed-off-by: Avik Basu <[email protected]>
numaproj · Jan 17, 2023 · a37d581 · a37d581
1 parent 48612fa
commit a37d581
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/Dockerfile b/Dockerfile
@@ -62,6 +62,7 @@ WORKDIR $PYSETUP_PATH
 COPY ./pyproject.toml ./poetry.lock ./
 RUN poetry install --no-cache --no-root && \
     poetry run pip install --no-cache torch --extra-index-url https://download.pytorch.org/whl/cpu && \
+    poetry run pip install pytorch-lightning && \
     rm -rf ~/.cache/pypoetry/
 
 ADD . /app

diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "numalogic-prometheus"
-version = "0.1.0a0"
+version = "0.1.0a1"
 description = "ML inference on numaflow using numalogic on Prometheus metrics"
 authors = ["Numalogic developers"]
 packages = [{ include = "numaprom" }]