gomoku_rl.utils.test module

class gomoku_rl.utils.test.Type(value)[source]

Bases: Enum

An enumeration.

black = 1

mixed = 3

white = 2

gomoku_rl.utils.test.assert_layer_transition(layer: Tensor, next_layer: Tensor, done: Tensor)[source]

gomoku_rl.utils.test.assert_observation(observation: Tensor, type: Type)[source]

gomoku_rl.utils.test.assert_tensor_1d_all(t: Tensor)[source]

gomoku_rl.utils.test.assert_transition(tensordict: TensorDict, type: Type)[source]

gomoku_rl.utils.test.no_nan_in_tensordict(tensordict: TensorDict)[source]