490f5b7364da85d414f4b8e93b3699e4d1b516f5,tests/garage/tf/baselines/test_continuous_mlp_baseline.py,TestContinuousMLPBaseline,test_is_pickleable,#TestContinuousMLPBaseline#,74

Before Change

    def test_is_pickleable(self):
        box_env = GarageEnv(DummyBoxEnv(obs_dim=(1, )))
        with mock.patch(("garage.tf.baselines."
                         "continuous_mlp_baseline."
                         "ContinuousMLPRegressor"),
                        new=SimpleMLPRegressor):
            cmb = ContinuousMLPBaseline(env_spec=box_env.spec)
        obs = {"observations": [np.full(1, 1), np.full(1, 1)]}

        with tf.compat.v1.variable_scope("ContinuousMLPBaseline", reuse=True):
            return_var = tf.compat.v1.get_variable("SimpleMLPModel/return_var")

After Change

        with tf.compat.v1.variable_scope("ContinuousMLPBaseline", reuse=True):
            bias = tf.compat.v1.get_variable("mlp/hidden_0/bias")
        bias.load(tf.ones_like(bias).eval())

        _, _, paths, _ = get_train_test_data()
        result1 = cmb.predict(paths)
        h = pickle.dumps(cmb)

Instances