diff --git a/deepchem/feat/graph_features.py b/deepchem/feat/graph_features.py index 82e01e37c0..6469645d08 100644 --- a/deepchem/feat/graph_features.py +++ b/deepchem/feat/graph_features.py @@ -690,9 +690,9 @@ def __init__(self, graph_distance=True, explicit_H=False, # If uses use_chirality self.use_chirality = use_chirality if self.use_chirality: - self.bt_len = bond_fdim_base + len(possible_bond_stereo) + self.bt_len = GraphConvConstants.bond_fdim_base + len(possible_bond_stereo) else: - self.bt_len = bond_fdim_base + self.bt_len = GraphConvConstants.bond_fdim_base def _featurize(self, mol): """Encodes mol as a WeaveMol object."""