def test_product_space_unflatten_n(): space = Product([Discrete(3), Discrete(3)]) np.testing.assert_array_equal(space.flatten((2, 2)), space.flatten_n([(2, 2)])[0]) np.testing.assert_array_equal( space.unflatten(space.flatten((2, 2))), space.unflatten_n(space.flatten_n([(2, 2)]))[0] )
def action_space(self): return Product([ Discrete(11), Discrete(11), Discrete(11), Discrete(2), Discrete(2) ])
def observation_space(self): """ Returns a Space object """ return Product(Discrete(52), # Player card 1 Discrete(52), # Player card 2 Discrete(52), # Player card 3 Discrete(52), # Dealer card 1 Discrete(52), # Dealer card 2 Discrete(52)) # Dealer card 3
def test_product_space(): _ = Product([Discrete(3), Discrete(2)]) product_space = Product(Discrete(3), Discrete(2)) sample = product_space.sample() assert product_space.contains(sample)