Skip to content

Commit

Permalink
collect-prec: add number of unmodelled residues as metadata columns
Browse files Browse the repository at this point in the history
  • Loading branch information
avivrosenberg committed Jun 13, 2024
1 parent d31b0ac commit 7ec2019
Showing 1 changed file with 7 additions and 0 deletions.
7 changes: 7 additions & 0 deletions src/pp5/collect.py
Original file line number Diff line number Diff line change
Expand Up @@ -58,6 +58,7 @@
COL_PDB_SOURCE = "pdb_source"
COL_REJECTED_BY = "rejected_by"
COL_NUM_ALTLOCS = "num_altlocs"
COL_NUM_UNMODELLED = "num_unmodelled"

COLLECTION_METADATA_FILENAME = "meta.json"
ALL_STRUCTS_FILENAME = "meta-structs_all"
Expand Down Expand Up @@ -1145,6 +1146,12 @@ def _collect_single_structure(
COL_SEQ_LEN: seq_len,
COL_SEQ_GAPS: str.join(";", [f"{s}-{e}" for (s, e) in prec.seq_gaps]),
COL_NUM_ALTLOCS: prec.num_altlocs,
**{
f"{COL_NUM_UNMODELLED}_{suffix}": n
for n, suffix in zip(
prec.num_unmodelled, ["nterm", "inter", "cterm"]
)
},
COL_PDB_SOURCE: pdb_source,
**meta.as_dict(chain_id=chain_id, seq_to_str=True),
}
Expand Down

0 comments on commit 7ec2019

Please sign in to comment.