From 53a80d5dc41666e755cf7c1198729dd950726f1c Mon Sep 17 00:00:00 2001
From: "pre-commit-ci[bot]"
 <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Date: Fri, 20 Dec 2024 22:18:15 +0000
Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
---
 grits/finegrain.py | 44 +++++++++++++++++++++++---------------------
 grits/utils.py     | 18 +++++++++++++++---
 2 files changed, 38 insertions(+), 24 deletions(-)

diff --git a/grits/finegrain.py b/grits/finegrain.py
index 7ab11f4..5d38892 100644
--- a/grits/finegrain.py
+++ b/grits/finegrain.py
@@ -13,24 +13,26 @@
 
 
 def backmap_snapshot_to_compound(
-        snapshot,
-        bead_mapping=None,
-        bond_head_index=None,
-        bond_tail_index=None,
-        library_key=None,
-        ref_distance=None,
-        energy_minimize=False
+    snapshot,
+    bead_mapping=None,
+    bond_head_index=None,
+    bond_tail_index=None,
+    library_key=None,
+    ref_distance=None,
+    energy_minimize=False,
 ):
     # TODO
     # assert all 3 dicts have the same keys
-    if (bead_mapping is None and bond_head_index is None and bond_tail_index is None) == (library_key is None):
-            raise ValueError(
-                "Please provide dictionaries or library key."
-            )
+    if (
+        bead_mapping is None
+        and bond_head_index is None
+        and bond_tail_index is None
+    ) == (library_key is None):
+        raise ValueError("Please provide dictionaries or library key.")
     if library_key is not None:
-        bead_mapping=reactant_dict[library_key]['smiles']
-        bond_tail_index=reactant_dict[library_key]['tail_indices']
-        bond_head_index=reactant_dict[library_key]['head_indices']
+        bead_mapping = reactant_dict[library_key]["smiles"]
+        bond_tail_index = reactant_dict[library_key]["tail_indices"]
+        bond_head_index = reactant_dict[library_key]["head_indices"]
     if not ref_distance:
         ref_distance = 1
     cg_snap = snapshot
@@ -47,11 +49,11 @@ def backmap_snapshot_to_compound(
     for mapping in bead_mapping:
         comp = mb.load(bead_mapping[mapping], smiles=True)
         if bond_head_index and bond_tail_index:
-            remove_atoms = [] # These will be removed
-            anchor_particles = [] # Store this for making bonds later
-            '''adding section to remove other particles in reacting group
-            assuming input for head/tail indices is a list, with the anchor particle listed first'''
-            if len(bond_tail_index[mapping]) > 1: 
+            remove_atoms = []  # These will be removed
+            anchor_particles = []  # Store this for making bonds later
+            """adding section to remove other particles in reacting group
+            assuming input for head/tail indices is a list, with the anchor particle listed first"""
+            if len(bond_tail_index[mapping]) > 1:
                 extra_tail_particles = []
                 extra_tail_particles = bond_tail_index[mapping][1:]
                 for k in extra_tail_particles:
@@ -136,8 +138,8 @@ def backmap_snapshot_to_compound(
             temp_head = temp_comp.children[1]
             tail_comp.xyz = temp_tail.xyz
             head_comp.xyz = temp_head.xyz
-            '''maybe replace this section with just:
-            fg_compound.energy_minimize(steps=500)'''
+            """maybe replace this section with just:
+            fg_compound.energy_minimize(steps=500)"""
     return fg_compound
 
 
diff --git a/grits/utils.py b/grits/utils.py
index 6e08c77..88209f2 100644
--- a/grits/utils.py
+++ b/grits/utils.py
@@ -476,7 +476,19 @@ def num2str(num):
 }
 
 reactant_dict = {
-    'pps':{'smiles':{"A":"c1ccc(S)cc1"},'head_indices':{"A":[7]}, 'tail_indices':{"A":[10]}},
-    'polystyrene':{'smiles':{"A":"C=CC1=CC=CC=C1"}, 'head_indices':{"A":[9]}, 'tail_indices':{"A":[10]}},
-    'polyalanine':{'smiles':{"A":"C[C@@H](C(=O)O)N"}, 'head_indices':{"A":[12]}, 'tail_indices':{"A":[4,10]}}
+    "pps": {
+        "smiles": {"A": "c1ccc(S)cc1"},
+        "head_indices": {"A": [7]},
+        "tail_indices": {"A": [10]},
+    },
+    "polystyrene": {
+        "smiles": {"A": "C=CC1=CC=CC=C1"},
+        "head_indices": {"A": [9]},
+        "tail_indices": {"A": [10]},
+    },
+    "polyalanine": {
+        "smiles": {"A": "C[C@@H](C(=O)O)N"},
+        "head_indices": {"A": [12]},
+        "tail_indices": {"A": [4, 10]},
+    },
 }