From 221fdc0774b1a65e802037d4d78b8a249353bb1e Mon Sep 17 00:00:00 2001
From: krfricke <krfricke@users.noreply.github.com>
Date: Tue, 11 Aug 2020 23:17:31 +0200
Subject: [PATCH] [tune] fix flaky PBT replay test (#10047)

Co-authored-by: Kai Fricke <kai@anyscale.com>
---
 python/ray/tune/tests/test_trial_scheduler.py | 15 ++++++++++++++-
 1 file changed, 14 insertions(+), 1 deletion(-)

diff --git a/python/ray/tune/tests/test_trial_scheduler.py b/python/ray/tune/tests/test_trial_scheduler.py
index d21f24d04..ad3d7f298 100644
--- a/python/ray/tune/tests/test_trial_scheduler.py
+++ b/python/ray/tune/tests/test_trial_scheduler.py
@@ -1075,11 +1075,24 @@ class PopulationBasedTestingSuite(unittest.TestCase):
         shutil.rmtree(tmpdir)
 
     def testReplay(self):
+        # Returns unique increasing parameter mutations
+        class _Counter:
+            def __init__(self, start=0):
+                self.count = start - 1
+
+            def __call__(self, *args, **kwargs):
+                self.count += 1
+                return self.count
+
         pbt, runner = self.basicSetup(
             num_trials=4,
             perturbation_interval=5,
             log_config=True,
-            step_once=False)
+            step_once=False,
+            hyperparam_mutations={
+                "float_factor": lambda: 100.0,
+                "int_factor": _Counter(1000)
+            })
         trials = runner.get_trials()
         tmpdir = tempfile.mkdtemp()