add dependencies

dmarx · Dec 9, 2023 · 66764cd · 66764cd
1 parent a159820
commit 66764cd
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/setup.py b/setup.py
@@ -21,6 +21,9 @@
         'sortedcontainers',
         'omegaconf',
         'matplotlib',
+        #############
+        'numpy',
+        'torch'
     ],
     extras_require={
         'dev': [