code: def _get_config(self, max_steps, lr_scheduler=False):
      config = {
          "trainer": {"params": {"max_steps": max_steps}},
          "training": {"lr_scheduler": lr_scheduler},
      }
      return get_config_with_defaults(config)
  
file_path: mmf/tests/trainers/lightning/test_lr_schedule.py
imports: import unittest
  from unittest.mock import MagicMock, patch
  import torch
  from mmf.trainers.callbacks.lr_scheduler import LRSchedulerCallback
  from tests.trainers.test_utils import get_config_with_defaults, get_lightning_trainer, get_mmf_trainer
label_desc: Use image node_id 3 for calling the TestLightningTrainerLRSchedule obj's underlying member method code with example usage: obj._get_config(max_steps, lr_scheduler) and returns: get_config_with_defaults