9a8724d219ce8758a652358c6578b09eb193a0df,catalyst/dl/experiment/config.py,ConfigExperiment,_get_scheduler,#,346

Before Change

        key_value_flag = params.pop("_key_value", False)

        if key_value_flag:
            scheduler = {}
            for scheduler_key, scheduler_params in params.items():
                scheduler[
                    scheduler_key
                ] = ConfigExperiment._get_scheduler(  // noqa: WPS437
                    optimizer=optimizer, **scheduler_params
                )
        else:
            scheduler = SCHEDULERS.get_from_params(
                **params, optimizer=optimizer
            )

After Change

        *, optimizer: Union[Optimizer, Dict[str, Optimizer]], **params: Any
    ) -> Union[Scheduler, Dict[str, Scheduler]]:
        optimizer_key = params.pop("_optimizer", None)
        optimizer_ = optimizer[optimizer_key] if optimizer_key else optimizer
        scheduler = SCHEDULERS.get_from_params(**params, optimizer=optimizer_)

        return scheduler

Instances