Mark flaky test

rlworkgroup · Mar 4, 2020 · be4f360 · be4f360
1 parent 7c45c35
commit be4f360
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/tests/garage/tf/algos/test_batch_polopt2.py b/tests/garage/tf/algos/test_batch_polopt2.py
@@ -1,6 +1,7 @@
 from unittest import mock
 
 import numpy as np
+import pytest
 
 from garage.np.baselines import LinearFeatureBaseline
 from garage.tf.algos import BatchPolopt2
@@ -18,6 +19,8 @@
 
 class TestBatchPolopt2(TfGraphTestCase):
 
+    # This test cause low memory with some reason
+    @pytest.mark.flaky
     @mock.patch.multiple(BatchPolopt2, __abstractmethods__=set())
     # pylint: disable=abstract-class-instantiated, no-member
     def test_process_samples_continuous_non_recurrent(self):