Commit dfdcc270 authored by peastman's avatar peastman
Browse files

Fixed error on Python 2

parent 50773098

deepchem/models/tensorgraph/tensor_graph.py

+1 −1
Original line number Diff line number Diff line
@@ -965,7 +965,7 @@ class TensorGraph(Model): 
    Any subclass that overrides default_generator() must also override this

    method.

    """

    if self.__class__.default_generator is not TensorGraph.default_generator:

    if self.__class__.default_generator != TensorGraph.default_generator:

      raise ValueError(

          "Class overrides default_generator() but not create_estimator_inputs()"

      )

CRA Git | Maintained and supported by SUSTech CRA and CCSE