fix: invalid split leaves selection variable undefined

[andrewwhitecdw]

Problem

fix: invalid split leaves selection variable undefined

Fix

Replace:

        if split is None:
            selection = slice(None)
        if split == 'train':
            # select 90% of all scenes
            selection = (pairs['seq_low'] % 10) > 0
        if split == 'val':
            # select 10% of all scenes
            selection = (pairs['seq_low'] % 10) == 0

with:

        if split is None:
            selection = slice(None)
        elif split == 'train':
            # select 90% of all scenes
            selection = (pairs['seq_low'] % 10) > 0
        elif split == 'val':
            # select 10% of all scenes
            selection = (pairs['seq_low'] % 10) == 0
        else:
            raise ValueError(f'bad {split=}')

Files changed

• dust3r/datasets/blendedmvs.py