Commit 7c369fe5 authored by Milosz Grabski's avatar Milosz Grabski
Browse files

pytest fix

parent 32301db3

deepchem/models/molgan.py

+3 −2
Original line number Diff line number Diff line
@@ -109,8 +109,9 @@ class BasicMolGANModel(WGAN): 
    List

        List of shapes used as an input for distriminator.

    """

    return [(self.vertices, self.vertices, self.edges), (self.vertices,

                                                         self.nodes)]

    return [(self.vertices, self.vertices, self.edges),

            (self.vertices, self.nodes), (self.vertices,

                                          self.vertices), (self.vertices)]



  def create_generator(self) -> keras.Model:

    """

CRA Git | Maintained and supported by SUSTech CRA and CCSE