code: def cleanup_data_file():
      script_directory = os.path.dirname(os.path.realpath(__file__))
      # List all files in the directory
      for name in TEST_CONFIG_FILES_NAME:
          name = name.split(".")[0]
          try:
              files = os.listdir(os.path.join(script_directory, name))
          except:
              return
          # Iterate over the files and remove the ones with .txt extension
          for file in files:
              if file.endswith(".txt"):
                  file_path = os.path.join(
                      script_directory + "/" + name, file
                  )
                  os.remove(file_path)
  
file_path: vowpal_wabbit/python/tests/e2e_v2/test_core.py
imports: from vw_executor.vw import Vw
  from vw_executor.vw_opts import Grid
  import pytest
  import os
  import logging
  from test_helper import json_to_dict_list, evaluate_expression, copy_file, custom_sort, get_function_obj_with_dirs, datagen_driver
  from conftest import STORE_OUTPUT
label_desc: Use image node_id 1 for calling a global function with example usage: cleanup_data_file() without return types