from typing import OrderedDict


def _sort_values_by_step(vals_dict):
    """Sorts values dictionary by step

    Args:
        vals_dict (dict): a values dictionary as created by extract_values

    Returns:
        dict: values dictionary with sorted steps
    """
    for k, v in vals_dict.items():
        vals_dict[k] = OrderedDict(sorted(v.items()))
    return vals_dict