code: def is_input_data_file(input_file):
      return "train-set" in input_file or "test-set" in input_file
  
imports: import copy
  import os
  import os.path
  import shlex
  import subprocess
  from pathlib import Path
  import re
  from typing import List, Optional, Tuple
  from run_tests_common import TestData
label_desc: Use image node_id 2 for calling a global function with example usage: is_input_data_file(input_file) and returns: unknown