def dynamics_update_tester(dynamics, observation_shape, action_size, discrete=False): transitions = base_update_tester(dynamics, observation_shape, action_size, discrete) # dummy algo algo = DummyAlgo(action_size, discrete) new_transitions = dynamics.generate(algo, transitions) assert len(new_transitions) == dynamics.horizon * dynamics.n_transitions
def algo_update_tester(algo, observation_shape, action_size, discrete=False): base_update_tester(algo, observation_shape, action_size, discrete)
def dynamics_update_tester( dynamics, observation_shape, action_size, discrete=False ): transitions = base_update_tester( dynamics, observation_shape, action_size, discrete )