import sys

from custom.dataset.vidar_dataset import VidarDataset 
from display.datasets.DisplayDataset import DisplayDataset


if __name__ == "__main__":

    # dataset_dir = 'custom/display/config/default.yaml'
    # dataset = VidarDataset(dataset_dir=dataset_dir, config_key=sys.argv[1])

    dataset_dir, config_key = sys.argv[1], 'dataset'
    dataset = VidarDataset(dataset_dir=dataset_dir, config_key=config_key)

    display = DisplayDataset(dataset.vidar_dataset.datasets[0])
    display.loop()