Function build_kwargs_batch_extractor

sam3d_objects/data/utils.py:135–142

(kwargs_mapping: KwargsType)

def build_kwargs_batch_extractor(kwargs_mapping: KwargsType):
    def extract_fn(batch):
        return {
            name: get_child(batch, *as_keys(path))
            for name, path in kwargs_mapping.items()
        }

    return extract_fn


empty_mapping = object()