From d296c9150840dd1380e8d82a6778eec93095877e Mon Sep 17 00:00:00 2001
From: wassname <wassname@users.noreply.github.com>
Date: Sat, 1 Dec 2018 17:14:20 +0800
Subject: [PATCH] fix env interface

---
 src/js/ddpg/ddpg_agent.js | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

diff --git a/src/js/ddpg/ddpg_agent.js b/src/js/ddpg/ddpg_agent.js
index 1d0ec15..bf4ada1 100644
--- a/src/js/ddpg/ddpg_agent.js
+++ b/src/js/ddpg/ddpg_agent.js
@@ -142,12 +142,10 @@ class DDPGAgent {
         const tfActions = this.ddpg.perturbedPrediction(tfPreviousStep);
         // Step in the environment with theses actions
         let mAcions = tfActions.buffer().values;
-        let mReward = this.env.step(mAcions);
+        let [mState, mReward, mDone, info] = this.env.step(mAcions);
         this.rewardsList.push(mReward);
         // Get the new observations
-        let mState = this.env.getState();
         let tfState = tf.tensor2d([mState]);
-        let mDone = 0;
         if (mReward == -1 && this.config.stopOnRewardError){
             mDone = 1;
         }