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

Add flaky annotation to test_wgan
JungeAlexander · Nov 5, 2017 · 548aa38 · 548aa38
2 parents c790a3b + 86bd351
commit 548aa38
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/deepchem/models/tensorgraph/tests/test_gan.py b/deepchem/models/tensorgraph/tests/test_gan.py
@@ -3,6 +3,7 @@
 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 @@ def create_discriminator(self, data_inputs, conditional_inputs):
     assert abs(np.mean(deltas)) < 1.0
     assert np.std(deltas) > 1.0
 
+  @flaky
   def test_wgan(self):
     """Test fitting a conditional WGAN."""