From 024d6961619f3bc061ada77991347922ce9690f3 Mon Sep 17 00:00:00 2001 From: adamjking3 <adamjking3@gmail.com> Date: Fri, 5 Jul 2019 20:49:51 -0700 Subject: [PATCH] Allow multi-process test to complete. --- lib/RLTrader.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/lib/RLTrader.py b/lib/RLTrader.py index 97b3ca9..36d3974 100644 --- 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)