Change

deepchem · Jul 11, 2020 · e8d4765 · e8d4765
1 parent b66503d
commit e8d4765
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 7 deletions.
diff --git a/deepchem/feat/graph_features.py b/deepchem/feat/graph_features.py
@@ -114,7 +114,7 @@ def safe_index(l, e):
     return len(l)
 
 
-class GraphConvConstants(enum.Enum):
+class GraphConvConstants(object):
   """Allowed Atom Types."""
   possible_atom_list = [
       'C', 'N', 'O', 'S', 'F', 'P', 'Cl', 'Mg', 'Na', 'Br', 'Fe', 'Ca', 'Cu',
@@ -690,9 +690,10 @@ 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 = GraphConvConstants.bond_fdim_base + len(possible_bond_stereo)
+      self.bt_len = int(
+          GraphConvConstants.bond_fdim_base) + len(possible_bond_stereo)
     else:
-      self.bt_len = GraphConvConstants.bond_fdim_base
+      self.bt_len = int(GraphConvConstants.bond_fdim_base)
 
   def _featurize(self, mol):
     """Encodes mol as a WeaveMol object."""

diff --git a/deepchem/models/graph_models.py b/deepchem/models/graph_models.py
@@ -430,10 +430,6 @@ def __init__(self,
       if dropout is None or dropout == 0.0:
         raise ValueError('Dropout must be included to predict uncertainty')
 
-    ############################################
-    print("self.dropout")
-    print(self.dropout)
-    ############################################
     # Build the model.
 
     atom_features = Input(shape=(self.n_atom_feat,))