Change

    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',

    # 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."""

      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,))