def _ray_function(): init_ray() env = ray_function() params = { "actions": { "dtype": numpy.int64 }, "critic": { "dtype": numpy.float32 } } states = StatesModel(state_dict=params, batch_size=N_WALKERS) return env, states
def swarm_with_score(self, request): init_ray() swarm = swarm_dict.get(request.param, create_cartpole_swarm)() score = test_scores[request.param] request.addfinalizer(lambda: kill_swarm(swarm)) return swarm, score
def export_swarm(request): init_ray() swarm = swarm_dict.get(request.param)() request.addfinalizer(lambda: kill_swarm(swarm)) return swarm
def export_swarm(request): init_ray() swarm = swarm_dict.get(request.param)() yield swarm kill_swarm(swarm)