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

for more information, see https://pre-commit.ci
apax-hub · Apr 3, 2024 · 844a393 · 844a393
1 parent 763bf14
commit 844a393
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
@@ -151,4 +151,4 @@ def forward(
             stress = grads[-1]
             prediction["stress"] = stress
 
-        return prediction
+        return prediction