code: def test_updates(self):
      with patch(
          "mmf.trainers.lightning_trainer.get_mmf_env", return_value=""
      ):
          config = self._get_config(max_steps=2, max_epochs=None)
          trainer = get_lightning_trainer(config=config)
          self.assertEqual(trainer._max_updates, 2)
  
          self._check_values(trainer, 0, 0)
          trainer.trainer.fit(
              trainer.model, trainer.data_module.train_loader
          )
          self._check_values(trainer, 2, 1)
  
file_path: mmf/tests/trainers/lightning/test_loop_conditions.py
imports: import unittest
  from unittest.mock import patch
  from tests.trainers.test_utils import get_config_with_defaults, get_lightning_trainer
label_desc: Use image node_id 3 for calling the TestLightningTrainer obj's underlying member method code with example usage: obj.test_updates() without return types