From a73f87e159d33e33eab5aca4ca6fcd015e3aec77 Mon Sep 17 00:00:00 2001 From: Niklas Gebauer <2736079+NiklasGebauer@users.noreply.github.com> Date: Fri, 2 Sep 2022 20:58:02 +0200 Subject: [PATCH] Fix error in AddOffsets The atomrefs should be added instead of subtracted here. --- src/schnetpack/transform/atomistic.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/schnetpack/transform/atomistic.py b/src/schnetpack/transform/atomistic.py index a13672853..cc7a09ee6 100644 --- a/src/schnetpack/transform/atomistic.py +++ b/src/schnetpack/transform/atomistic.py @@ -227,6 +227,6 @@ def forward( if not self.is_extensive: y0 /= inputs[structure.n_atoms] - inputs[self._property] -= y0 + inputs[self._property] += y0 return inputs