ruff

Genentech · Apr 25, 2024 · 5b08e94 · 5b08e94
1 parent 2e94b82
commit 5b08e94
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/beignet/func/_molecular_dynamics/_space/_space.py b/src/beignet/func/_molecular_dynamics/_space/_space.py
@@ -107,6 +107,7 @@ def space(
         dimensions = torch.tensor([dimensions])
 
     if dimensions is None:
+
         def _displacement_fn(
             a: Tensor,
             b: Tensor,
@@ -134,6 +135,7 @@ def _shift_fn(a: Tensor, b: Tensor, **_) -> Tensor:
         inverse_transformation = _inverse_transform(dimensions)
 
         if normalized:
+
             def _displacement_fn(
                 a: Tensor,
                 b: Tensor,
@@ -171,6 +173,7 @@ def _displacement_fn(
                 return displacement
 
             if remapped:
+
                 def _u(a: Tensor, b: Tensor) -> Tensor:
                     return torch.remainder(a + b, 1.0)
 
@@ -252,6 +255,7 @@ def _displacement_fn(
             return displacement
 
         if remapped:
+
             def _u(a: Tensor, b: Tensor) -> Tensor:
                 return torch.remainder(a + b, 1.0)
 
@@ -308,9 +312,11 @@ 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