3cbeab42c246425425c584cd53f1cc25d8c738a7,torchreid/datasets/__init__.py,,init_vidreid_dataset,#Any#,52

Before Change

def init_vidreid_dataset(name, **kwargs):
    if name not in list(__vidreid_factory.keys()):
        raise KeyError("Invalid dataset, got "{}", but expected to be one of {}".format(name, list(__vidreid_factory.keys())))
    return __vidreid_factory[name](**kwargs)

After Change

def init_vidreid_dataset(name, **kwargs):
    avai_datasets = list(__vidreid_factory.keys())
    if name not in avai_datasets:
        raise RuntimeError("Invalid dataset name. Received "{}", but expected to be one of {}".format(name, avai_datasets))
    return __vidreid_factory[name](**kwargs)

Instances