def test_tf_worker_without_default_session(self): tf_worker = TFWorkerWrapper() worker = DefaultWorker(seed=1, max_path_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
def test_tf_worker_with_default_session(self): with LocalTFRunner(snapshot_config): tf_worker = TFWorkerWrapper() worker = DefaultWorker(seed=1, max_episode_length=100, worker_number=1) worker.update_env(GymEnv(DummyBoxEnv())) tf_worker._inner_worker = worker tf_worker.worker_init() assert tf_worker._sess == tf.compat.v1.get_default_session() assert tf_worker._sess._closed