Function pprint_data_mixture

data/utils/data_utils.py:193–207

(
    dataset_kwargs_list: List[Dict[str, Any]], dataset_weights: List[int]
)

# === RLDS Dataset Initialization Utilities ===
def pprint_data_mixture(
    dataset_kwargs_list: List[Dict[str, Any]], dataset_weights: List[int]
) -> None:
    print(
        "\n######################################################################################"
    )
    print(
        f"# Loading the following {len(dataset_kwargs_list)} datasets (incl. sampling weight):{'': >24} #"
    )
    for dataset_kwargs, weight in zip(dataset_kwargs_list, dataset_weights):
        pad = 80 - len(dataset_kwargs["name"])
        print(f"# {dataset_kwargs['name']}: {weight:=>{pad}f} #")
    print(
        "######################################################################################\n"
    )


def cartesian_to_spherical(x, y, z):