It seems like the SHAWrapFedex missed to take client_cfgs. Please check.
The Line
|
elif init_cfg.hpo.scheduler.startswith('wrap', client_cfgs): |
elif init_cfg.hpo.scheduler.startswith('wrap', client_cfgs):
scheduler = SHAWrapFedex(init_cfg)
perhaps to be changed to
elif init_cfg.hpo.scheduler.startswith('wrap'):
scheduler = SHAWrapFedex(init_cfg, client_cfgs)