def get_available_datasets():
    """Find all unique dataset names across all checkpoints"""
    datasets = set()
    for checkpoint in os.listdir(DATA_DIR):
        if checkpoint.startswith(CHECKPOINT_PREFIX):
            checkpoint_path = os.path.join(DATA_DIR, checkpoint)
            if os.path.isdir(checkpoint_path):
                datasets.update(d for d in os.listdir(checkpoint_path))
    return sorted(datasets)
Find all unique dataset names across all checkpoints
examples/simplelr_math_eval/visualizer.py