Formatting.

Author: ameya98

src/e3tools/nn/_extract_irreps.py

@@ -29,7 +29,7 @@ def forward(self, data: torch.Tensor) -> torch.Tensor:
 
         Parameters:
             data: torch.Tensor of shape [..., irreps_in.dim]
-        
+
         Returns:
             torch.Tensor of shape [..., irreps_out.dim]
         """

src/e3tools/nn/_mlp.py

@@ -9,7 +9,7 @@
 
 class ScalarMLP(torch.nn.Sequential):
     """A multi-layer perceptron for scalar inputs and outputs."""
-    
+
     def __init__(
         self,
         in_features: int,

src/e3tools/nn/_pack_unpack.py

@@ -99,7 +99,7 @@ def forward(self, x: torch.Tensor) -> torch.Tensor:
 
         Parameters:
             x: torch.Tensor of shape [..., irreps.dim]
-        
+
         Returns:
             torch.Tensor of shape [..., factor, irreps.dim // factor]
         """
@@ -121,7 +121,7 @@ def forward(self, x: torch.Tensor) -> torch.Tensor:
 
         Parameters:
             x: torch.Tensor of shape [..., factor, irreps.dim // factor]
-        
+
         Returns:
             torch.Tensor of shape [..., irreps.dim]
         """

src/e3tools/nn/_scaling.py

@@ -21,7 +21,7 @@ def forward(self, data: torch.Tensor, weights: torch.Tensor) -> torch.Tensor:
         Parameters:
             data: torch.Tensor of shape [..., irreps_in.dim]
             weights: torch.Tensor of shape [..., irreps_in.num_irreps]
-        
+
         Returns:
             torch.Tensor of shape [..., irreps_in.dim]
         """