code: def enable_tqdm_ray(request):
      context = ray.data.DataContext.get_current()
      original_use_ray_tqdm = context.use_ray_tqdm
      context.use_ray_tqdm = request.param
      yield request.param
      context.use_ray_tqdm = original_use_ray_tqdm
  
imports: import functools
  from pytest import fixture
  import ray
  from ray.data._internal.progress_bar import ProgressBar
label_desc: Use image node_id 1 for calling a global function with example usage: enable_tqdm_ray(request) without return types