From 4454d05bcfe654cc93aa8edc99c5dd48d575eca0 Mon Sep 17 00:00:00 2001 From: Amog Kamsetty Date: Mon, 13 Jul 2020 13:15:01 -0700 Subject: [PATCH] [Tune] Trainable documentation fix (#9448) --- python/ray/tune/trainable.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/ray/tune/trainable.py b/python/ray/tune/trainable.py index 545f88385..f6bfa8e2c 100644 --- a/python/ray/tune/trainable.py +++ b/python/ray/tune/trainable.py @@ -175,7 +175,7 @@ class Trainable: Calling ``save()`` should save the training state of a trainable to disk, and ``restore(path)`` should restore a trainable to the given state. - Generally you only need to implement ``build``, ``step``, + Generally you only need to implement ``setup``, ``step``, ``save_checkpoint``, and ``load_checkpoint`` when subclassing Trainable. Other implementation methods that may be helpful to override are