code: def get_files(input_path: Path) -> List[str]:
      supported_file_types = ["jsonl.zst", ".txt", ".xz", ".tar.gz"]
      if input_path.is_dir():
          # get all files with supported file types
          files = [
              list(Path(input_path).glob(f"*{ft}"))
              for ft in supported_file_types
          ]
          # flatten list
          files = [f for sublist in files for f in sublist]
          assert (
              files
          ), f"No files with supported types found in directory: {input_path}"
      elif input_path.is_file():
          assert any(
              str(input_path).endswith(f_type)
              for f_type in supported_file_types
          ), f"Input file type must be one of: {supported_file_types}"
          files = [input_path]
      else:
          raise FileNotFoundError(
              f"No such file or directory: {input_path=}"
          )
  
      return [str(f) for f in files]
  
imports: import argparse
  import os
  import re
  import random
  from pathlib import Path
  from typing import List
  import ftfy
  import tensorflow
  from lm_dataformat import Reader
  from transformers import GPT2TokenizerFast
  from tqdm import tqdm
label_desc: Use image node_id 2 for calling a global function with example usage: get_files(input_path) and returns: unknown