[RLlib] use Mapping instead of dict in summarize() to accommodate non-dict grads/params (e.g. haiku's frozendict) (#8793)

This commit is contained in:
Kristian Holsheimer
2020-06-11 17:37:15 +10:00
committed by GitHub
parent 700d81fa20
commit ea965d7c52
+2 -1
View File
@@ -1,5 +1,6 @@
import numpy as np
import pprint
from typing import Mapping
from ray.rllib.policy.sample_batch import SampleBatch, MultiAgentBatch
@@ -17,7 +18,7 @@ def summarize(obj):
def _summarize(obj):
if isinstance(obj, dict):
if isinstance(obj, Mapping):
return {k: _summarize(v) for k, v in obj.items()}
elif hasattr(obj, "_asdict"):
return {