INNER CODE UNIT · Python

__getitem__

allenai/ai2thor · ai2thor/server.py:34

    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):
        pass

    def __iter__(self):
        self._load_all()

View source record →