diff --git a/ogb/utils/features.py b/ogb/utils/features.py index 442173b6..4ffb53cf 100644 --- a/ogb/utils/features.py +++ b/ogb/utils/features.py @@ -6,6 +6,11 @@ 'CHI_TETRAHEDRAL_CW', 'CHI_TETRAHEDRAL_CCW', 'CHI_OTHER', + 'CHI_TETRAHEDRAL', + 'CHI_ALLENE', + 'CHI_SQUAREPLANAR', + 'CHI_TRIGONALBIPYRAMIDAL', + 'CHI_OCTAHEDRAL', 'misc' ], 'possible_degree_list' : [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 'misc'],