# Autogenerated via:
# python ~/code/watch/dev/maintain/mirror_package_geowatch.py


def __getattr__(key):
    import watch.tasks.rutgers_material_change_detection.datasets.iarpa_sc_kwdataset as mirror
    return getattr(mirror, key)


def __dir__():
    import watch.tasks.rutgers_material_change_detection.datasets.iarpa_sc_kwdataset as mirror
    return dir(mirror)


if __name__ == '__main__':
    # Get root directory of dataset on local machine.
    from watch.tasks.rutgers_material_change_detection.utils.util_paths import get_dataset_root_dir
    from watch.tasks.rutgers_material_change_detection.utils.util_misc import generate_video_slice_object

    dset_name = "iarpa_drop2v2"
    root_dir = get_dataset_root_dir(dset_name)

    # Get dataset split and task_mode.
    split = "valid"
    task_mode = "total_sem_change"

    # Test loading first and last frames of training split.
    video_slice = generate_video_slice_object(height=300, width=300, n_frames=5, scale=1)
    dset = IARPA_SC_EVAL_DATASET(root_dir, split, video_slice, task_mode, channels="RGB_NIR", normalize_mode=None)
    example = dset.__getitem__(0)
