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

for more information, see https://pre-commit.ci
a-r-j · Oct 28, 2023 · d6647be · d6647be
1 parent 45b020e
commit d6647be
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/graphein/protein/utils.py b/graphein/protein/utils.py
@@ -417,12 +417,8 @@ def save_graph_to_pdb(
     for col in blank_cols:
         if col not in df.columns:
             df[col] = ""
-    atom_df = filter_dataframe(
-        df, "record_name", ["ATOM"], boolean=True
-    )
-    hetatm_df = filter_dataframe(
-        df, "record_name", ["HETATM"], boolean=True
-    )
+    atom_df = filter_dataframe(df, "record_name", ["ATOM"], boolean=True)
+    hetatm_df = filter_dataframe(df, "record_name", ["HETATM"], boolean=True)
 
     if atoms:
         ppd.df["ATOM"] = atom_df