Add missing dependencies to setup.py

bfirsh · bfirsh · commit ee37e02885ff · 2021-09-28T16:45:07.000-07:00
diff --git a/setup.py b/setup.py
@@ -9,5 +9,7 @@
         'torch',
         'numpy',
         'tqdm',
+        'pytorch-lightning',
+        'einops'
     ],
 )
