add cuequivariance-torch to dependencies

ACEsuit · Nov 21, 2024 · 5b2bd9a · 5b2bd9a
1 parent a200571
commit 5b2bd9a
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/setup.cfg b/setup.cfg
@@ -29,6 +29,7 @@ install_requires =
     GitPython
     pyYAML
     tqdm
+    cuequivariance-torch
     # for plotting:
     matplotlib
     pandas
@@ -59,3 +60,5 @@ dev =
     pytest-benchmark
     pylint
 schedulefree = schedulefree
+cueq-cuda-11 = cuequivariance-ops-torch-cu11
+cueq-cuda-12 = cuequivariance-ops-torch-cu12