def test_bcmodule_rnn_dc_update(is_sac):
mock_brain = mb.create_mock_banana_brain()
bc_module = create_bc_module(mock_brain, ppo_dummy_config(), True,
"testdcvis.demo", is_sac)
stats = bc_module.update()
for _, item in stats.items():
assert isinstance(item, np.float32)
def test_bcmodule_rnn_dc_update(trainer_config):
mock_brain = mb.create_mock_banana_brain()
policy = create_policy_with_bc_mock(mock_brain, trainer_config, True,
"testdcvis.demo")
stats = policy.bc_module.update()
for _, item in stats.items():
assert isinstance(item, np.float32)
def test_bcmodule_dc_visual_update(mock_env, trainer_config):
mock_brain = mb.create_mock_banana_brain()
env, policy = create_policy_with_bc_mock(mock_env, mock_brain,
trainer_config, False,
"testdcvis.demo")
stats = policy.bc_module.update()
for _, item in stats.items():
assert isinstance(item, np.float32)
env.close()
def test_bcmodule_rnn_dc_update(is_sac):
mock_brain = mb.create_mock_banana_brain()
bc_settings = BehavioralCloningSettings(
demo_path=os.path.dirname(os.path.abspath(__file__)) + "/" +
"testdcvis.demo")
bc_module = create_bc_module(mock_brain, bc_settings, True, is_sac)
stats = bc_module.update()
for _, item in stats.items():
assert isinstance(item, np.float32)
