Unverified Commit 548aa380 authored by Bharath Ramsundar's avatar Bharath Ramsundar Committed by GitHub
Browse files

Merge pull request #923 from deepchem/rbharath-patch-2 

Add flaky annotation to test_wgan
parents c790a3b2 86bd3518

deepchem/models/tensorgraph/tests/test_gan.py

+2 −0
Original line number Diff line number Diff line
@@ -3,6 +3,7 @@ import numpy as np 
import tensorflow as tf

import unittest

from deepchem.models.tensorgraph import layers

from flaky import flaky





def generate_batch(batch_size):
@@ -58,6 +59,7 @@ class TestGAN(unittest.TestCase): 
    assert abs(np.mean(deltas)) < 1.0

    assert np.std(deltas) > 1.0



  @flaky

  def test_wgan(self):

    """Test fitting a conditional WGAN."""

CRA Git | Maintained and supported by SUSTech CRA and CCSE