fix indentation

choderalab · Jan 31, 2024 · ec3c0c2 · ec3c0c2
1 parent ffa571c
commit ec3c0c2
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/mtenn/conversion_utils/visnet.py b/mtenn/conversion_utils/visnet.py
@@ -89,9 +89,8 @@ def forward(self, data):
             x, v = self.visnet.representation_model(z, pos, batch)
             x = self.visnet.output_model.pre_reduce(x, v)
             x = x * self.visnet.std
-
-        if self.visnet.prior_model is not None:
-            x = self.visnet.prior_model(x, z)
+            if self.visnet.prior_model is not None:
+                x = self.visnet.prior_model(x, z)
 
             return x