Commit 48054f24 authored by leswing's avatar leswing
Browse files

Flaky also

parent ae210dbf
Loading
Loading
Loading
Loading
+3 −0
Original line number Diff line number Diff line
from flaky import flaky

import deepchem as dc
from deepchem.models.tensorgraph.layers import Reshape, Variable, SoftMax
import numpy as np
@@ -7,6 +9,7 @@ import unittest

class TestA3C(unittest.TestCase):

  @flaky
  def test_roulette(self):
    """Test training a policy for the roulette environment."""