def parse_json_stats()

in pycls/utils/logging.py [0:0]

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

def parse_json_stats(log, row_type, key):
    """Extract values corresponding to row_type/key out of log."""
    vals = [row[key] for row in log if row['_type'] == row_type and key in row]
    if key == 'iter' or key == 'epoch':
        vals = [int(val.split('/')[0]) for val in vals]
    return vals