From c683485b648d1db6a8f1d9a52801f02e3b660681 Mon Sep 17 00:00:00 2001 From: chungongyu Date: Mon, 9 Feb 2026 11:05:30 +0800 Subject: [PATCH] feat: add atom_type_num for NA and PROT --- profold2/common/residue_constants.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/profold2/common/residue_constants.py b/profold2/common/residue_constants.py index 1c249ff4..a35b00d7 100644 --- a/profold2/common/residue_constants.py +++ b/profold2/common/residue_constants.py @@ -859,6 +859,10 @@ def make_bond_key(atom1_name, atom2_name): atom_type_num = len(atom_types) # := 37. atom14_type_num = env('profold2_atom14_type_num', defval=14, dtype=int) # := 14. +prot_atom_type_num = 14 +dna_atom_type_num = 22 +rna_atom_type_num = 23 + # A compact atom encoding with 14 columns # pylint: disable=line-too-long # pylint: disable=bad-whitespace