From ea965d7c52afde1e1953ac3bc0aea3673f2dd5ab Mon Sep 17 00:00:00 2001
From: Kristian Holsheimer <kristian.holsheimer@gmail.com>
Date: Thu, 11 Jun 2020 17:37:15 +1000
Subject: [PATCH] [RLlib] use Mapping instead of dict in summarize() to
 accommodate non-dict grads/params (e.g. haiku's frozendict) (#8793)

---
 rllib/utils/debug.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/rllib/utils/debug.py b/rllib/utils/debug.py
index 3e44d67fc..24a09302d 100644
--- a/rllib/utils/debug.py
+++ b/rllib/utils/debug.py
@@ -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 {