space

Genentech · Apr 22, 2024 · f1a1258 · f1a1258
1 parent 774afbf
commit f1a1258
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/src/beignet/func/_molecular_dynamics/_space/_space.py b/src/beignet/func/_molecular_dynamics/_space/_space.py
@@ -188,9 +188,7 @@ def _shift_fn(a: Tensor, b: Tensor, **kwargs) -> Tensor:
                     if "transformation" in kwargs:
                         _transformation = kwargs["transformation"]
 
-                        _inverse_transformation = _inverse_transform(
-                            _transformation
-                        )
+                        _inverse_transformation = _inverse_transform(_transformation)
 
                     if "updated_transformation" in kwargs:
                         _transformation = kwargs["updated_transformation"]
@@ -325,4 +323,4 @@ def _shift_fn(a: Tensor, b: Tensor, **_) -> Tensor:
         def _shift_fn(a: Tensor, b: Tensor, **_) -> Tensor:
             return a + b
 
-    return _displacement_fn, _shift_fn
+    return _displacement_fn, _shift_fn