def test_integer(self):
     space = Integer(lower=1, upper=4)
     for i in range(4):
         self.assertEqual(space[i], i + 1)
     self.assertEqual("Integer(lower=1, upper=4, default=None)", str(space))
     self.assertEqual(get_max(space), 4)
     self.assertEqual(get_min(space), 1)
Esempio n. 2
0
 def out_features(self):
     return spaces.get_max(self._out_features)
Esempio n. 3
0
 def in_features(self):
     return spaces.get_max(self._in_features)
Esempio n. 4
0
 def hidden_multiplier(self):
     return spaces.get_max(self._hidden_multiplier)
 def d_model(self):
     return spaces.get_max(self._d_model)
Esempio n. 6
0
 def in_dim(self):
     return spaces.get_max(self._in_dim)
Esempio n. 7
0
 def embed_dim(self):
     return spaces.get_max(self._embed_dim)
 def proj_dim(self):
     return spaces.get_max(self._proj_dim)
 def num_heads(self):
     return spaces.get_max(self._num_heads)