Commit cfe2ad80 authored by leswing's avatar leswing
Browse files

function signature

parent e2bc6a63

deepchem/models/tensorgraph/models/atomic_conv.py

+1 −1
Original line number Diff line number Diff line
@@ -64,7 +64,7 @@ class AtomicConvScore(Layer): 
    self.layer_sizes = layer_sizes

    super(AtomicConvScore, self).__init__(**kwargs)



  def create_tensor(self, **kwargs):

  def create_tensor(self, in_layers=None, set_tensors=True, **kwargs):

    frag1_layer = self.in_layers[0].out_tensor

    frag2_layer = self.in_layers[1].out_tensor

    complex_layer = self.in_layers[2].out_tensor

CRA Git | Maintained and supported by SUSTech CRA and CCSE