(input_dir)
return names
get all file names located in dir_path
get all file names located in dir_path
get all file names located in dir_path
def get_file_names(input_dir):
    """
        get all file names located in dir_path
    """
    sub_name = 'data_batch'
    files = os.listdir(input_dir)
    names = [each_item for each_item in files if sub_name in each_item]
    return names
def get_file_names ( input_dir ) : sub_name = 'data_batch' files = os . listdir ( input_dir ) names = [ each_item for each_item in files if sub_name in each_item ] return names