docs: document on_trial_close estimator hook (#683)

(cherry picked from commit 2f8f2c6)
determined-ai · Jun 12, 2020 · 97a26a2 · 97a26a2
1 parent d770579
commit 97a26a2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/docs/reference/api/estimator.txt b/docs/reference/api/estimator.txt
@@ -56,7 +56,7 @@ To execute arbitrary Python functionality during the lifecycle of a
 `tf.estimator.SessionRunHook <https://www.tensorflow.org/api_docs/python/tf/estimator/SessionRunHook/>`_.
 
 .. autoclass:: determined.estimator.RunHook
-    :members: on_checkpoint_load, on_checkpoint_end
+    :members: on_checkpoint_load, on_checkpoint_end, on_trial_close
 
 
 Example usage of ``determined.estimator.RunHook`` which adds custom metadata checkpoints: