def get_tfp_distrib_params(
    distrib,
    valid_params = {
        'concentration',
        'loc',
        'scale',
        'df',
        'total_count',
        'covariance_matrix',
    },
):
    """Returns the actual distribution parameters of the tfp distribution."""
    return {k: v for k, v in distrib._parameters.items() if k in valid_params}