b9b54d4f30ff65cf1c54dc0cf90c938b48c44f90,reagent/test/models/test_dqn.py,TestFullyConnectedDQN,test_save_load,#TestFullyConnectedDQN#,34

Before Change

    def test_save_load(self):
        state_dim = 8
        action_dim = 4
        model = FullyConnectedDQN(
            action_dim=action_dim,
            embedding=FullyConnectedNetwork(
                layers=[state_dim, 8, 4],
                activations=["relu", "relu"],
                use_batch_norm=True,
            ),
        )
        expected_num_params, expected_num_inputs, expected_num_outputs = 6, 1, 1
        check_save_load(
            self, model, expected_num_params, expected_num_inputs, expected_num_outputs
        )

After Change

    def test_save_load(self):
        state_dim = 8
        action_dim = 4
        model = FullyConnectedDQN(
            state_dim,
            action_dim,
            sizes=[8, 4],
            activations=["relu", "relu"],
            use_batch_norm=False,
        )
        expected_num_params, expected_num_inputs, expected_num_outputs = 6, 1, 1
        check_save_load(
            self, model, expected_num_params, expected_num_inputs, expected_num_outputs
        )

Instances