From e276a8bf97ab60bd3d62b6c6cab301be180b1f53 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Tong=20Guo=20=28=E9=83=AD=E5=90=8C=29?= <779222056@qq.com>
Date: Mon, 11 Oct 2021 12:06:35 +0800
Subject: [PATCH] Update trpo_mpi.py

---
 baselines/trpo_mpi/trpo_mpi.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/baselines/trpo_mpi/trpo_mpi.py b/baselines/trpo_mpi/trpo_mpi.py
index cd1e7eab77..e197d79632 100644
--- a/baselines/trpo_mpi/trpo_mpi.py
+++ b/baselines/trpo_mpi/trpo_mpi.py
@@ -1,11 +1,12 @@
-from baselines.common import explained_variance, zipsame, dataset
+from baselines.common import dataset
 from baselines import logger
 import baselines.common.tf_util as U
 import tensorflow as tf, numpy as np
 import time
 from baselines.common import colorize
 from collections import deque
-from baselines.common import set_global_seeds
+from baselines.common.misc_util import set_global_seeds, zipsame
+from baselines.common.math_util import explained_variance
 from baselines.common.mpi_adam import MpiAdam
 from baselines.common.cg import cg
 from baselines.common.input import observation_placeholder