[tune] fix example in wait_for_gpu docstring

Signed-off-by: Kevin Su <[email protected]>
sihanwang41 · Jan 2, 2024 · 3a15b56 · 3a15b56
1 parent 9bb973a
commit 3a15b56
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/ray/tune/utils/util.py b/python/ray/tune/utils/util.py
@@ -475,7 +475,7 @@ def wait_for_gpu(
  .. code-block:: python
 
  def tune_func(config):
- tune.util.wait_for_gpu()
+ tune.utils.wait_for_gpu()
  train()
 
  tuner = tune.Tuner(