8295d05ccbe3d5aaa07440f2409cb8fa69ec88fc,tests/garage/tf/baselines/test_gaussian_mlp_baseline.py,TestGaussianMLPBaseline,test_is_pickleable,#TestGaussianMLPBaseline#,76

Before Change

                         "GaussianMLPRegressor"),
                        new=SimpleGaussianMLPRegressor):
            gmb = GaussianMLPBaseline(env_spec=box_env.spec)
        obs = {"observations": [np.full(1, 1), np.full(1, 1)]}

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

After Change

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

        prediction = gmb.predict(paths)

        h = pickle.dumps(gmb)

Instances