3a3bdf17ca08052309016bdfa481a58c70112abd,tests/garage/tf/policies/test_gaussian_mlp_policy_with_model.py,TestGaussianMLPPolicyWithModel,test_is_pickleable,#TestGaussianMLPPolicyWithModel#Any#Any#,93

Before Change

        p = pickle.dumps(policy)
        with tf.Session(graph=tf.Graph()):
            policy_pickled = pickle.loads(p)
            action2, prob2 = policy_pickled.get_action(obs)

        assert env.action_space.contains(action1)
        assert np.array_equal(action1, action2)

After Change

        obs, _, _, _ = env.step(1)
        obs_dim = env.spec.observation_space.flat_dim

        with tf.variable_scope(
                "GaussianMLPPolicyWithModel/GaussianMLPModel", reuse=True):
            return_var = tf.get_variable("return_var")
        // assign it to all one
        return_var.load(tf.ones_like(return_var).eval())
        output1 = self.sess.run(
            policy.model.outputs[:-1],
            feed_dict={policy.model.input: [obs.flatten()]})

Instances