mirror of
https://github.com/wassname/ray.git
synced 2026-07-07 06:39:58 +08:00
[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:
committed by
GitHub
parent
700d81fa20
commit
ea965d7c52
@@ -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 {
|
||||
|
||||
Reference in New Issue
Block a user