From 5b2bd9a821be8e3ac3b0531af5999fa3fe7e510a Mon Sep 17 00:00:00 2001
From: Ilyes Batatia <48651863+ilyes319@users.noreply.github.com>
Date: Thu, 21 Nov 2024 11:26:58 +0000
Subject: [PATCH] add cuequivariance-torch to dependencies

---
 setup.cfg | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/setup.cfg b/setup.cfg
index ba714d85..be4804c4 100644
--- 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