def list_tf_records()

in spotify_tensorflow/scripts/tfr_read.py [0:0]

• 7 lines of code
• 6 McCabe index (conditional complexity)

def list_tf_records(paths, default_schema):
    for p in paths:
        files = [f for f in file_io.get_matching_files(p) if f.endswith(".tfrecords")]
        if len(files) == 0:
            raise Exception("Couldn't find any .tfrecords file in path or glob [{}]".format(p))
        for f in files:
            yield f, resolve_schema(os.path.dirname(f), default_schema)