diff --git a/doc/source/data/api/api.rst b/doc/source/data/api/api.rst index e0b6df24ac30a..b82d011b1d125 100644 --- a/doc/source/data/api/api.rst +++ b/doc/source/data/api/api.rst @@ -12,6 +12,5 @@ Ray Data API execution_options.rst grouped_data.rst data_context.rst - utility.rst preprocessor.rst from_other_data_libs.rst diff --git a/doc/source/data/api/utility.rst b/doc/source/data/api/utility.rst deleted file mode 100644 index 29e5d67271f38..0000000000000 --- a/doc/source/data/api/utility.rst +++ /dev/null @@ -1,12 +0,0 @@ -.. _data-utility: - -Utility -======= - -.. currentmodule:: ray.data - -.. autosummary:: - :nosignatures: - :toctree: doc/ - - set_progress_bars diff --git a/python/ray/data/__init__.py b/python/ray/data/__init__.py index 58514478a9ae3..89d531aa2ee55 100644 --- a/python/ray/data/__init__.py +++ b/python/ray/data/__init__.py @@ -12,7 +12,6 @@ NodeIdStr, ) from ray.data._internal.logging import configure_logging -from ray.data._internal.progress_bar import set_progress_bars from ray.data.context import DataContext, DatasetContext from ray.data.dataset import Dataset, Schema from ray.data.datasource import ( @@ -151,7 +150,6 @@ "read_sql", "read_tfrecords", "read_webdataset", - "set_progress_bars", "Preprocessor", "TFXReadOptions", ] diff --git a/python/ray/data/_internal/progress_bar.py b/python/ray/data/_internal/progress_bar.py index 2d794bfa7cea3..805a0724ca89f 100644 --- a/python/ray/data/_internal/progress_bar.py +++ b/python/ray/data/_internal/progress_bar.py @@ -5,7 +5,6 @@ import ray from ray.experimental import tqdm_ray from ray.types import ObjectRef -from ray.util.annotations import Deprecated from ray.util.debug import log_once logger = logging.getLogger(__name__) @@ -23,24 +22,6 @@ _canceled_threads_lock = threading.Lock() -@Deprecated -def set_progress_bars(enabled: bool) -> bool: - """Set whether progress bars are enabled. - - The default behavior is controlled by the - ``RAY_DATA_DISABLE_PROGRESS_BARS`` environment variable. By default, - it is set to "0". Setting it to "1" will disable progress bars, unless - they are reenabled by this method. - - Returns: - Whether progress bars were previously enabled. - """ - raise DeprecationWarning( - "`set_progress_bars` is deprecated. Set " - "`ray.data.DataContext.get_current().enable_progress_bars` instead.", - ) - - def extract_num_rows(result: Any) -> int: """Extract the number of rows from a result object. diff --git a/python/ray/data/tests/test_progress_bar.py b/python/ray/data/tests/test_progress_bar.py index 039ca8ee10e8c..abaaa0c0a1575 100644 --- a/python/ray/data/tests/test_progress_bar.py +++ b/python/ray/data/tests/test_progress_bar.py @@ -18,11 +18,6 @@ def enable_tqdm_ray(request): context.use_ray_tqdm = original_use_ray_tqdm -def test_set_progress_bars_is_deprecated(): - with pytest.raises(DeprecationWarning): - ray.data.set_progress_bars(True) - - def test_progress_bar(enable_tqdm_ray): total = 10 # Used to record the total value of the bar at close