fix env bug

WellyZhang · Nov 14, 2016 · d9a5ec1 · d9a5ec1
1 parent 30d45a0
commit d9a5ec1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/run.py b/run.py
@@ -1,4 +1,4 @@
-from rllab.algos.ddpg import DDPG
+from ddpg import DDPG
 from rllab.envs.box2d.cartpole_env import CartpoleEnv
 from rllab.envs.normalized_env import normalize
 from policies import DeterministicMLPPolicy
@@ -8,7 +8,7 @@
 
 # set environment, policy, qfunc, strategy
 
-env = normalize(CartpoleEnv)
+env = normalize(CartpoleEnv())
 
 policy = DeterministicMLPPolicy(env.spec)
 qfunc = ContinuousMLPQ(env.spec)