code: def generate_data(
      num_rows: int,
      num_files: int,
      num_row_groups_per_file: int,
      compression: str,
      data_dir: str,
  ):
      results = []
      for file_index, global_row_index in enumerate(
          range(0, num_rows, num_rows // num_files)
      ):
          num_rows_in_file = min(
              num_rows // num_files, num_rows - global_row_index
          )
          results.append(
              generate_file.remote(
                  file_index,
                  global_row_index,
                  num_rows_in_file,
                  num_row_groups_per_file,
                  compression,
                  data_dir,
              )
          )
      filenames, data_sizes = zip(*ray.get(results))
      return filenames, sum(data_sizes)
  
imports: import os
  import pandas
  import numpy
  import ray
label_desc: Use image node_id 1 for calling a global function with example usage: generate_data(num_rows, num_files, num_row_groups_per_file, compression, data_dir) and returns: filenames,  sum