Skip to content

Commit

Permalink
[pre-commit.ci] auto fixes from pre-commit.com hooks
Browse files Browse the repository at this point in the history
for more information, see https://pre-commit.ci
  • Loading branch information
pre-commit-ci[bot] committed Apr 14, 2024
1 parent cae7aa4 commit 3da6e06
Showing 1 changed file with 4 additions and 2 deletions.
6 changes: 4 additions & 2 deletions graphein/protein/tensor/io.py
Original file line number Diff line number Diff line change
Expand Up @@ -108,7 +108,7 @@ def protein_to_pyg(
atom_types: List[str] = PROTEIN_ATOMS,
remove_nonstandard: bool = True,
store_het: bool = False,
fill_value_coords: float = 1e-5
fill_value_coords: float = 1e-5,
) -> Data:
"""
Parses a protein (from either: a PDB code, PDB file or a UniProt ID
Expand Down Expand Up @@ -238,7 +238,9 @@ def protein_to_pyg(
df["residue_id"] = df.residue_id + ":" + df.insertion

out = Data(
coords=protein_df_to_tensor(df, atoms_to_keep=atom_types, fill_value=fill_value_coords),
coords=protein_df_to_tensor(
df, atoms_to_keep=atom_types, fill_value=fill_value_coords
),
residues=get_sequence(
df,
chains=chain_selection,
Expand Down

0 comments on commit 3da6e06

Please sign in to comment.