From c1c5ffd6f8b2b520c4f174d9d187dddc99c41966 Mon Sep 17 00:00:00 2001 From: niuyazhe Date: Fri, 22 Dec 2023 16:11:35 +0800 Subject: [PATCH] fix(nyz): fix comm unittest bug --- ding/worker/learner/comm/base_comm_learner.py | 1 + 1 file changed, 1 insertion(+) diff --git a/ding/worker/learner/comm/base_comm_learner.py b/ding/worker/learner/comm/base_comm_learner.py index a7234e0e7a..4a9562888f 100644 --- a/ding/worker/learner/comm/base_comm_learner.py +++ b/ding/worker/learner/comm/base_comm_learner.py @@ -116,6 +116,7 @@ def _create_learner(self, task_info: dict) -> 'BaseLearner': # noqa setattr(learner, item, getattr(self, item)) # Set policy in created learner. policy_cfg = task_info['policy'] + policy_cfg = EasyDict(policy_cfg) learner.policy = create_policy(policy_cfg, enable_field=['learn']).learn_mode learner.setup_dataloader() return learner