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)