[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
apax-hub · Apr 3, 2024 · 956d816 · 956d816
1 parent c16c29a
commit 956d816
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/apax/nn/torch/model/gmnn.py b/apax/nn/torch/model/gmnn.py
@@ -93,4 +93,4 @@ def forward(
         #     energy_correction = correction(dr_vec, Z, idx)
         #     total_energy = total_energy + energy_correction
 
-        return total_energy
+        return total_energy