def test_interval(self):
"""Check whether interval is cool."""
space = Space()
probs = (0.1, 0.2, 0.3, 0.4)
categories = ('asdfa', 2, 3, 4)
dim = Categorical('yolo', OrderedDict(zip(categories, probs)), shape=2)
space.register(dim)
dim = Integer('yolo2', 'uniform', -3, 6)
space.register(dim)
dim = Real('yolo3', 'norm', 0.9)
space.register(dim)
assert space.interval() == [categories, (-3, 3), (-np.inf, np.inf)]
def __init__(self, roi_space: Space, n_levels: int):
self.n_levels = n_levels
self.space = roi_space
self.sk_space = SkSpace(fix_shape_intervals(roi_space.interval()))
self.sk_space.set_transformer("normalize")
