Write out pdb70_hits.hhr alongside other MSA outputs.

PiperOrigin-RevId: 386894639 Change-Id: I1ef2018a48398cf8201ac38fe129b004672ad22f

Write out pdb70_hits.hhr alongside other MSA outputs.
PiperOrigin-RevId: 386894639 Change-Id: I1ef2018a48398cf8201ac38fe129b004672ad22f
2028cd0a · DeepMind · Copybara-Service · 20a70577 · 2028cd0a
Commit 2028cd0a authored Jul 26, 2021 by DeepMind Committed by Copybara-Service Jul 26, 2021
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 0 deletions

alphafold/data/pipeline.py alphafold/data/pipeline.py +4 -0

No files found.
--- a/alphafold/data/pipeline.py
+++ b/alphafold/data/pipeline.py
@@ -148,6 +148,10 @@ class DataPipeline:
    with open(mgnify_out_path, 'w') as f:
      f.write(jackhmmer_mgnify_result['sto'])

+    pdb70_out_path = os.path.join(msa_output_dir, 'pdb70_hits.hhr')
+    with open(pdb70_out_path, 'w') as f:
+      f.write(hhsearch_result)
+
    uniref90_msa, uniref90_deletion_matrix, _ = parsers.parse_stockholm(
        jackhmmer_uniref90_result['sto'])
    mgnify_msa, mgnify_deletion_matrix, _ = parsers.parse_stockholm(