From 17c0cc7fadcb15ac6d4d828f8eeff7aaf35678d6 Mon Sep 17 00:00:00 2001
From: Andrina Zhang <78346223+andrinazxx@users.noreply.github.com>
Date: Tue, 28 May 2024 20:03:55 -0700
Subject: [PATCH] Update Dockerfile - lightning doesn't work

---
 Dockerfile | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/Dockerfile b/Dockerfile
index 0b86de3..116c275 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -21,7 +21,7 @@ USER jovyan
 
 # RUN conda install -y scikit-learn
 
-RUN pip install --no-cache-dir networkx scipy wave soundfile pytorch-lightning
+RUN pip install --no-cache-dir networkx scipy wave soundfile pytorch-lightning pytorch_lightning
 
 # Override command to disable running jupyter notebook at launch
 # CMD ["/bin/bash"]