コード例 #1
0
ファイル: test_tf_worker.py プロジェクト: songanz/garage
 def test_tf_worker_without_default_session(self):
     tf_worker = TFWorkerWrapper()
     worker = DefaultWorker(seed=1, max_episode_length=100, worker_number=1)
     worker.update_env(DummyBoxEnv())
     tf_worker._inner_worker = worker
     tf_worker.worker_init()
     assert tf_worker._sess == tf.compat.v1.get_default_session()
     tf_worker.shutdown()
     assert tf_worker._sess._closed