Add missing dependencies to setup.py

bfirsh · bfirsh · commit 9036ea08453c · 2021-11-10T08:15:38.000-08:00
diff --git a/setup.py b/setup.py
@@ -9,5 +9,7 @@
         'torch',
         'numpy',
         'tqdm',
+        'pytorch-lightning',
+        'einops'
     ],
 )
