diff --git a/mace/modules/radial.py b/mace/modules/radial.py index 57f106d0..cf626c78 100644 --- 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: