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 20, 2024
1 parent b4bf164 commit 6bfb02e
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions graphein/protein/tensor/io.py
Original file line number Diff line number Diff line change
Expand Up @@ -163,7 +163,7 @@ def protein_to_pyg(
:param store_bfactor: Whether or not to store bfactors in the ``Data``
object. Default is ``False.
:type store_bfactor: bool
:param fill_value_coords: Fill value to use for positions in atom37
:param fill_value_coords: Fill value to use for positions in atom37
representation that are not filled. Defaults to 1e-5
:type fill_value_coords: float
:returns: ``Data`` object with attributes: ``x`` (AtomTensor), ``residues``
Expand Down Expand Up @@ -261,7 +261,7 @@ def protein_to_pyg(
)
if store_het:
out.hetatms = [het_coords]

if store_bfactor:
out.bfactor = torch.from_numpy(df["b_factor"].values)

Expand Down

0 comments on commit 6bfb02e

Please sign in to comment.