From 905d12bfa2884e2228f8299c0bfca450daa45d3a Mon Sep 17 00:00:00 2001 From: haoshengzou Date: Fri, 9 Mar 2018 09:25:19 +0800 Subject: [PATCH] working on tester --- tianshou/data/tester.py | 26 +++++++++++++++++++++++--- 1 file changed, 23 insertions(+), 3 deletions(-) diff --git a/tianshou/data/tester.py b/tianshou/data/tester.py index 2a2e407..2f21def 100644 --- a/tianshou/data/tester.py +++ b/tianshou/data/tester.py @@ -1,8 +1,28 @@ from __future__ import absolute_import +import gym +import logging + + +def test_policy_in_env(policy, env, num_episodes, num_timesteps=0): + + assert sum([num_episodes > 0, num_timesteps > 0]) == 1, \ + 'One and only one collection number specification permitted!' -def test_policy_in_env(policy, env): # make another env as the original is for training data collection - env_ = env + env_id = env.spec.id + env_ = gym.make(env_id) + # current_observation = env_.reset() - pass \ No newline at end of file + # test policy + if num_episodes > 0: + pass + + if num_timesteps > 0: + pass + + # log + + + # clear scene + env_.close()