def _load(self, output_fn, **kwargs):
    if (self._dataset_cls is not None):
        return self._dataset_cls(output_fn, **kwargs)
    elif self._mixed_dataset_types:
        return load(output_fn, **kwargs)
    ds = load(output_fn, **kwargs)
    self._dataset_cls = ds.__class__
    return ds
def _load(self, output_fn, **kwargs):
    if (self._dataset_cls is not None):
        return self._dataset_cls(output_fn, **kwargs)
    elif self._mixed_dataset_types:
        return load(output_fn, **kwargs)
    ds = load(output_fn, **kwargs)
    self._dataset_cls = ds.__class__
    return ds