Unwrap Coordinates Class

class mdsuite.transformations.unwrap_coordinates.CoordinateUnwrapper[source]

Unwrap coordinates by checking if particles moved from one side of the box to the other within one time step.

transform_batch(batch: Dict[str, Tensor], carryover: Optional[Any] = None) Tuple[Tensor, dict][source]

Implement parent class abstract method.