[tune] Fix SearchAlg finishing early (#3081)

* Fix trial search alg finishing early * Fix lint * fix lint * nit fix
2026-08-11 11:24:51 +08:00 · 2018-10-22 12:17:13 -07:00
parent 221d1663c1
commit eff7cb4458
3 changed files with 32 additions and 2 deletions
@@ -223,7 +223,7 @@ For TensorFlow model training, this would look something like this `(full tensor
 .. code-block:: python

    class MyClass(Trainable):
-        def _setup(self):
+        def _setup(self, config):
            self.saver = tf.train.Saver()
            self.sess = ...
            self.iteration = 0