INNER CODE UNIT · Python

LazyMask

allenai/ai2thor · ai2thor/server.py:30

class LazyMask(Mapping):
    def __contains__(self, key: object) -> bool:
        return self.mask(cast(str, key)) is not None

    def __getitem__(self, key: str):
        m = self.mask(key)

        if m is None:
            raise KeyError(key)

        return m

    @abstractmethod
    def mask(self, key: str, default: Optional[np.ndarray] = None) -> Optional[np.ndarray]:
        pass

    @abstractmethod
    def _load_all(self):

View source record →