Formatting updates

This commit is contained in:
Kevin Wu
2022-11-17 10:35:32 -08:00
parent 70542630fe
commit dd96b1a9a9

View File

@@ -36,6 +36,7 @@ import esm.inverse_folding
from biotite.structure.io.pdb import PDBFile
from biotite.sequence import ProteinSequence, AlphabetError
def get_chain_from_pdb(fname: str) -> str:
"""
Get the chain from the given PDB file. If multiple chains are present,
@@ -51,6 +52,7 @@ def get_chain_from_pdb(fname: str) -> str:
assert len(chain_ids) == 1
return chain_ids.pop()
def write_fa(fname: str, seq: str, seqname: str = "sampled"):
"""Write a fasta file"""
assert fname.endswith(".fasta")