Skip to content

Commit

Permalink
line length fixes
Browse files Browse the repository at this point in the history
  • Loading branch information
VarunAnanth2003 committed Nov 3, 2024
1 parent d581944 commit 092fa2a
Show file tree
Hide file tree
Showing 3 changed files with 11 additions and 6 deletions.
5 changes: 3 additions & 2 deletions casanovo/casanovo.py
Original file line number Diff line number Diff line change
Expand Up @@ -62,8 +62,9 @@ def __init__(self, *args, **kwargs) -> None:
click.Option(
("-m", "--model"),
help="""
Either the model weights (.ckpt file) or a URL pointing to the model weights
file. If not provided, Casanovo will try to download the latest release automatically.
Either the model weights (.ckpt file) or a URL pointing to
the model weights file. If not provided,
Casanovo will try to download the latest release automatically.
""",
),
click.Option(
Expand Down
4 changes: 2 additions & 2 deletions casanovo/denovo/model.py
Original file line number Diff line number Diff line change
Expand Up @@ -1021,8 +1021,8 @@ def predict_step(self, batch, *args):
predictions: List[Tuple[List[str], int, float, str, np.ndarray, np.ndarray, str]]
Model predictions for the given batch of spectra containing spectrum
ids, precursor charge and m/z, candidate peptide sequences, peptide
scores, amino acid-level scores, and associated proteins. Stored separately by
spectrum id.
scores, amino acid-level scores, and associated proteins.
Stored separately by spectrum id.
"""
store_dict = collections.defaultdict(list)
for start_idx in range(0, len(batch[0]), self.psm_batch_size):
Expand Down
8 changes: 6 additions & 2 deletions tests/conftest.py
Original file line number Diff line number Diff line change
Expand Up @@ -22,14 +22,18 @@ def tiny_fasta_file(tmp_path):
fasta_file = tmp_path / "tiny_fasta.fasta"
with fasta_file.open("w+") as fasta_ref:
fasta_ref.write(
">foo\nMEAPAQLLFLLLLWLPDTTREIVMTQSPPTLSLSPGERVTLSCRASQSVSSSYLTWYQQKPGQAPRLLIYGASTRATSIPARFSGSGSGTDFTLTISSLQPEDFAVYYCQQDYNLP"
(
">foo\nMEAPAQLLFLLLLWLPDTTREIVMTQSPPTLSLSPGERVTLSCRASQSVSSSYLTWYQ"
"QKPGQAPRLLIYGASTRATSIPARFSGSGSGTDFTLTISSLQPEDFAVYYCQQDYNLP"
)
)
return fasta_file


@pytest.fixture
def mgf_medium(tmp_path):
"""An MGF file with 7 spectra and scan numbers, C+57.021 mass modification considered"""
"""An MGF file with 7 spectra and scan numbers,
C+57.021 mass modification considered"""
peptides = [
"ATSIPAR",
"VTLSCR",
Expand Down

0 comments on commit 092fa2a

Please sign in to comment.