import sys

from vidar.utils.config import read_config
from vidar.utils.setup import setup_dataset
from display.datasets.DisplayDataset import DisplayDataset

cfg = read_config(f'utils/display/config.yaml')
dataset = cfg.datasets.dict[sys.argv[1]]

dataset = setup_dataset(dataset, verbose=True)[0][0]
display = DisplayDataset(dataset)
display.loop()