From d47d6a6b7aa5ab195fd9b01b9da8880e52e07aea Mon Sep 17 00:00:00 2001
From: Alok Singh <8325708+alok@users.noreply.github.com>
Date: Mon, 11 Jun 2018 09:53:31 -0700
Subject: [PATCH] [rllib] Use correct method name (#2226)

---
 python/ray/rllib/utils/tf_policy_graph.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/python/ray/rllib/utils/tf_policy_graph.py b/python/ray/rllib/utils/tf_policy_graph.py
index 6588060bf..a3dfd174b 100644
--- a/python/ray/rllib/utils/tf_policy_graph.py
+++ b/python/ray/rllib/utils/tf_policy_graph.py
@@ -105,7 +105,7 @@ class TFPolicyGraph(PolicyGraph):
         feed_dict[self._is_training] = True
         for ph, value in zip(self._grads, gradients):
             feed_dict[ph] = value
-        fetches = self.sess.run(
+        fetches = self._sess.run(
             [self._apply_op, self.extra_apply_grad_fetches()],
             feed_dict=feed_dict)
         return fetches[1]