[tune] Use public methods for trainable (#9184)

2026-08-06 13:31:10 +08:00 · 2020-07-01 11:00:00 -07:00
parent 1491508859
commit d35f0e40d0
40 changed files with 350 additions and 220 deletions
@@ -145,7 +145,7 @@ class RayTrialExecutorTest(unittest.TestCase):
        """Tests that reset works as expected."""

        class B(Trainable):
-            def _train(self):
+            def step(self):
                return dict(timesteps_this_iter=1, done=True)

            def reset_config(self, config):