fix the jit compile for radial

ACEsuit · Jan 15, 2025 · 83bfd9d · 83bfd9d
1 parent badc690
commit 83bfd9d
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/mace/modules/radial.py b/mace/modules/radial.py
@@ -129,6 +129,7 @@ def __init__(self, r_max: float, p=6):
     def forward(self, x: torch.Tensor) -> torch.Tensor:
         return self.calculate_envelope(x, self.r_max, self.p.to(torch.int))
 
+    @staticmethod
     def calculate_envelope(
         x: torch.Tensor, r_max: torch.Tensor, p: torch.Tensor
     ) -> torch.Tensor: