code: def __enter__(self):
      self.o._get_lr_called_within_step = True
      return self
  
imports: import torch
  from packaging.version import Version
label_desc: Use image node_id 2 for calling the _enable_get_lr_call obj's underlying member method code with example usage: obj.__enter__() and returns: self