stats

ray-project · Jun 5, 2019 · 1efb0c3 · 1efb0c3
1 parent eea2099
commit 1efb0c3
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/python/ray/rllib/policy/dynamic_tf_policy.py b/python/ray/rllib/policy/dynamic_tf_policy.py
@@ -337,6 +337,7 @@ def gen_loss(model_outputs, *args):
                 self.action_dist.inputs = model_outputs
                 return self._loss_fn(self, eager_inputs)
 
+            # TODO(ekl) also handle the stats funcs
             loss = tf.py_function(
                 gen_loss,
                 # cast works around TypeError: Cannot convert provided value