def unwrap()

in support/gym_remote/bridge.py [0:0]

• 15 lines of code
• 7 McCabe index (conditional complexity)

    def unwrap(space):
        if space.annotations['type'] == 'MultiBinary':
            return gym.spaces.MultiBinary(int(space.annotations['n']))
        if space.annotations['type'] == 'Discrete':
            return gym.spaces.Discrete(int(space.annotations['n']))
        if space.annotations['type'] == 'MultiDiscrete':
            if gym_version >= (0, 9, 6):
                return gym.spaces.MultiDiscrete(space.shape[0])
            else:
                return gym.spaces.MultiDiscrete(space.shape)
        if space.annotations['type'] == 'Box':
            kwargs = {}
            if gym_version >= (0, 9, 6):
                kwargs['dtype'] = space.dtype
            return gym.spaces.Box(low=0, high=255, shape=space.shape, **kwargs)