hiro/ray
1
0
Fork 0
mirror of https://github.com/vale981/ray synced 2025-03-06 10:31:39 -05:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fixes empty state argument in compute_single_action method (#6894)

Browse source 
* Fixes empty `state` parameter in compute_single_action method

* Fixed style
This commit is contained in:
AnanthHari 2020-01-23 03:42:52 -05:00 • committed by Eric Liang
parent 1558307ac4
commit aa2a0cb6da
1 changed files with 4 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
rllib/policy/policy.py
View file

@ -128,6 +128,7 @@ class Policy(metaclass=ABCMeta):
prev_reward_batch = None
info_batch = None
episodes = None
state_batch = None
if prev_action is not None:
prev_action_batch = [prev_action]
if prev_reward is not None:
@ -136,9 +137,11 @@ class Policy(metaclass=ABCMeta):
info_batch = [info]
if episode is not None:
episodes = [episode]
if state is not None:
state_batch = [[s] for s in state]
[action], state_out, info = self.compute_actions(
[obs], [[s] for s in state],
[obs], state_batch,
prev_action_batch=prev_action_batch,
prev_reward_batch=prev_reward_batch,
info_batch=info_batch,