update requirements

mathLab · Nov 2, 2023 · 1c49710 · 1c49710
1 parent 723d778
commit 1c49710
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -15,7 +15,7 @@
 KEYWORDS = 'physics-informed neural-network'
 
 REQUIRED = [
-    'numpy', 'matplotlib', 'torch', 'lightning'
+    'numpy', 'matplotlib', 'torch', 'lightning', 'pytorch_lightning'
 ]
 
 EXTRAS = {