a849c17e46ea7188a7b6229c9f68417803e730ad,rl_coach/architectures/tensorflow_components/shared_variables.py,SharedRunningStats,push,#SharedRunningStats#Any#,76

Before Change

        self.sess = sess

    def push(self, x):
        x = x.astype("float64")
        self.sess.run([self._inc_sum, self._inc_sum_squared, self._inc_count],
                         feed_dict={
                             self.new_sum: x.sum(axis=0).ravel(),
                             self.new_sum_squared: np.square(x).sum(axis=0).ravel(),
                             self.newcount: np.array(len(x), dtype="float64")
                         })
        if self._shape is None:
            self._shape = x.shape

    @property

After Change

    def push(self, x):
        if self.pubsub:
            self.pubsub.redis_connection.publish(self.channel, pickle.dumps(x))
            return

        self.push_val(x)

    def push_val(self, x):

Instances