INNER CODE UNIT · Python

NumpyAwareEncoder

allenai/ai2thor · ai2thor/server.py:21

class NumpyAwareEncoder(json.JSONEncoder):
    def default(self, obj):
        if isinstance(obj, np.ndarray):
            return obj.tolist()
        if isinstance(obj, np.generic):
            return obj.item()
        return super(NumpyAwareEncoder, self).default(obj)


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)

View source record →