def METHOD_NAME(dataset: Union[dict, pd.DataFrame]) -> Sequence[str]:
    if isinstance(dataset, pd.DataFrame):
        keys = dataset.columns.unique(level=0)
    else:
        # In case it is a dict
        keys = dataset.keys()
    return keys