[rllib] Native support for Dict and Tuple spaces; fix Tuple action spaces; add prev a, r to LSTM (#3051)

This commit is contained in:
Eric Liang
2018-10-20 15:21:22 -07:00
committed by GitHub
parent 9a2b5333ef
commit 59901a88a0
39 changed files with 921 additions and 185 deletions
+8 -2
View File
@@ -2,9 +2,11 @@ from __future__ import absolute_import
from __future__ import division
from __future__ import print_function
from collections import namedtuple
import distutils.version
import tensorflow as tf
import numpy as np
import distutils.version
use_tf150_api = (distutils.version.LooseVersion(tf.VERSION) >=
distutils.version.LooseVersion("1.5.0"))
@@ -225,4 +227,8 @@ class MultiActionDistribution(ActionDistribution):
def sample(self):
"""Draw a sample from the action distribution."""
return [[s.sample() for s in self.child_distributions]]
return TupleActions([s.sample() for s in self.child_distributions])
TupleActions = namedtuple("TupleActions", ["batches"])