revert e3nn version

ACEsuit · Oct 1, 2024 · 3ec597e · 3ec597e
1 parent 264817c
commit 3ec597e
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/mace/modules/blocks.py b/mace/modules/blocks.py
@@ -625,7 +625,7 @@ def _setup(self) -> None:
         )
 
         # Linear
-        # irreps_mid = irreps_mid.simplify()
+        irreps_mid = irreps_mid.simplify()
         self.irreps_out = self.target_irreps
         self.linear = o3.Linear(
             irreps_mid, self.irreps_out, internal_weights=True, shared_weights=True

diff --git a/setup.cfg b/setup.cfg
@@ -15,7 +15,7 @@ packages = find:
 python_requires = >=3.7
 install_requires =
     torch>=1.12
-    e3nn
+    e3nn==0.4.4
     numpy<2.0
     opt_einsum
     ase