Prettier code formatting

chaidiscovery · Dec 9, 2024 · f735d8b · f735d8b
1 parent 9d776b9
commit f735d8b
Showing 1 changed file with 12 additions and 8 deletions.
diff --git a/chai_lab/data/dataset/structure/all_atom_structure_context.py b/chai_lab/data/dataset/structure/all_atom_structure_context.py
@@ -171,10 +171,12 @@ def drop_leaving_atoms(self) -> None:
             bonds_b = [
                 candidate
                 for candidate in bond_candidates_b.tolist()
-                if self._infer_bonds_within_conformer(
-                    candidate, exclude_polymers=True
-                ).sum()
-                == 1
+                if (
+                    self._infer_bonds_within_conformer(
+                        candidate, exclude_polymers=True
+                    ).sum()
+                    == 1
+                )
             ]
             # If there are multiple such bonds, we can't infer which to drop
             if len(bonds_b) == 1:
@@ -195,10 +197,12 @@ def drop_leaving_atoms(self) -> None:
             bonds_a = [
                 candidate
                 for candidate in bond_candidates_a.tolist()
-                if self._infer_bonds_within_conformer(
-                    candidate, exclude_polymers=True
-                ).sum()
-                == 1
+                if (
+                    self._infer_bonds_within_conformer(
+                        candidate, exclude_polymers=True
+                    ).sum()
+                    == 1
+                )
             ]
             # If there are multiple such bonds, we can't infer which to drop
             if len(bonds_a) == 1: