Allow multi-process test to complete.

notadamking · Jul 6, 2019 · 024d696 · 024d696
1 parent 5146326
commit 024d696
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/lib/RLTrader.py b/lib/RLTrader.py
@@ -223,10 +223,11 @@ def test(self, model_epoch: int = 0, should_render: bool = True):
         self.logger.info(f'Testing model ({self.study_name}__{model_epoch})')
 
         state = None
-        obs, done, rewards = test_env.reset(), [False], []
-        while not all(done):
+        obs, rewards = test_env.reset(), []
+
+        for _ in range(len(test_provider.data_frame)):
             action, state = model.predict(obs, state=state)
-            obs, reward, done, _ = test_env.step(action)
+            obs, reward, _, __ = test_env.step(action)
 
             rewards.append(reward)