3cbeab42c246425425c584cd53f1cc25d8c738a7,torchreid/datasets/__init__.py,,init_imgreid_dataset,#Any#,46

Before Change

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


def init_vidreid_dataset(name, **kwargs):

After Change

def init_imgreid_dataset(name, **kwargs):
    avai_datasets = list(__imgreid_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 __imgreid_factory[name](**kwargs)


def init_vidreid_dataset(name, **kwargs):

Instances