message: Reduce scope of with statement
diff: @@ -123,8 +123,6 @@ def load_splits(
  local_datasets_dir = get_local_datasets_dir()
  - with concurrent.futures.ThreadPoolExecutor() as executor:
  -
  def load_method(split_id: str):
  data = load_split(
  dataset_name=dataset_name,
  @@ -136,10 +134,10 @@ def load_splits(
  return split_id, data
  + with concurrent.futures.ThreadPoolExecutor() as executor:
  future_results = [
  executor.submit(load_method, split_id) for split_id in dataset_split_hashes
  ]
  -
  concurrent.futures.wait(future_results)
  for future in future_results:
  