Commit 648f3616 authored by Bharath Ramsundar's avatar Bharath Ramsundar
Browse files

Fixing missed test

parent b5453eac

deepchem/models/tensorgraph/graph_layers.py

+1 −1
Original line number Diff line number Diff line
@@ -392,7 +392,7 @@ class DTNNStep(Layer): 
        self.W_cf, self.W_df, self.W_fc, self.b_cf, self.b_df

    ]



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

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

    """description and explanation refer to deepchem.nn.DTNNStep

    parent layers: atom_features, distance, distance_membership_i, distance_membership_j

    """

CRA Git | Maintained and supported by SUSTech CRA and CCSE