fix type in calculate_envelope

ACEsuit · Jan 15, 2025 · badc690 · badc690
1 parent c7c0229
commit badc690
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/mace/modules/radial.py b/mace/modules/radial.py
@@ -129,9 +129,8 @@ 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: int
+        x: torch.Tensor, r_max: torch.Tensor, p: torch.Tensor
     ) -> torch.Tensor:
         r_over_r_max = x / r_max
         envelope = (