fix the jit compile for radial

Author: ilyes319

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: