File size: 440 Bytes
714cf46 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 | from dataclasses import dataclass
from typing import List
@dataclass
class ProteinStructureTemplate:
sequence: str
residue_names: List[str]
atom_names: List[str]
atom_elements: List[str]
atom_residue_index: List[int]
atom_chain_id: List[str]
@property
def num_atoms(self) -> int:
return len(self.atom_names)
@property
def num_residues(self) -> int:
return len(self.residue_names)
|