wehub-resource-sync
26 lines
619 B
Python
26 lines
619 B
Python
"""This file is injected for Ray Data doctest targets."""
|
|
import pytest
|
|
|
|
import ray
|
|
|
|
|
|
@pytest.fixture(autouse=True, scope="module")
|
|
def shutdown_ray():
|
|
ray.shutdown()
|
|
yield
|
|
|
|
|
|
@pytest.fixture(autouse=True)
|
|
def preserve_block_order():
|
|
ray.data.context.DataContext.get_current().execution_options.preserve_order = True
|
|
yield
|
|
|
|
|
|
@pytest.fixture(autouse=True)
|
|
def disable_start_message():
|
|
context = ray.data.context.DataContext.get_current()
|
|
original_value = context.print_on_execution_start
|
|
context.print_on_execution_start = False
|
|
yield
|
|
context.print_on_execution_start = original_value
|