diff --git a/torchmdnet/priors/atomref.py b/torchmdnet/priors/atomref.py
index f2114bfaa..4e415b1f0 100644
--- a/torchmdnet/priors/atomref.py
+++ b/torchmdnet/priors/atomref.py
@@ -96,7 +96,6 @@ def pre_reduce(
             return x
 
 
-
 class LearnableAtomref(Atomref):
     r"""LearnableAtomref prior model.