Skip to content

Commit

Permalink
fix(nyz): fix comm unittest bug
Browse files Browse the repository at this point in the history
  • Loading branch information
PaParaZz1 committed Dec 22, 2023
1 parent cfbd7ea commit c1c5ffd
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions ding/worker/learner/comm/base_comm_learner.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit c1c5ffd

Please sign in to comment.