reformatted, docs fix, removed transorms

src/beignet/func/_molecular_dynamics/_space/_space.py

@@ -8,9 +8,6 @@
 import torch
 from torch import Tensor
 
-from .__inverse_transform import _inverse_transform
-from .__transform import _transform
-from ._transform import transform
 
 T = TypeVar("T")
 
@@ -43,7 +40,7 @@ def space(
         affine transformation, $T$, specified in one of three ways: a cube,
         $L$; an orthorhombic unit cell, $[L_{x}, L_{y}, L_{z}]$; or a triclinic
         cell, upper triangular matrix. If `parallelepiped` is `False`, must be
-        the edge lengths. If `None`, the simulation space has free boudnary
+        the edge lengths. If `None`, the simulation space has free boundary
         conditions.
 
     normalized : bool, default=True
@@ -110,7 +107,6 @@ def space(
         dimensions = torch.tensor([dimensions])
 
     if dimensions is None:
-
         def _displacement_fn(
             a: Tensor,
             b: Tensor,
@@ -138,7 +134,6 @@ def _shift_fn(a: Tensor, b: Tensor, **_) -> Tensor:
         inverse_transformation = _inverse_transform(dimensions)
 
         if normalized:
-
             def _displacement_fn(
                 a: Tensor,
                 b: Tensor,
@@ -176,7 +171,6 @@ def _displacement_fn(
                 return displacement
 
             if remapped:
-
                 def _u(a: Tensor, b: Tensor) -> Tensor:
                     return torch.remainder(a + b, 1.0)
 
@@ -258,7 +252,6 @@ def _displacement_fn(
             return displacement
 
         if remapped:
-
             def _u(a: Tensor, b: Tensor) -> Tensor:
                 return torch.remainder(a + b, 1.0)
 
@@ -315,11 +308,9 @@ def _displacement_fn(
         return displacement
 
     if remapped:
-
         def _shift_fn(a: Tensor, b: Tensor, **_) -> Tensor:
             return torch.remainder(a + b, dimensions)
     else:
-
         def _shift_fn(a: Tensor, b: Tensor, **_) -> Tensor:
             return a + b