code: def _test_lamb(self, steps, opts, atol, rtol):
      self._test_optim(
          LAMB, tfa.optimizers.LAMB, steps, opts, atol, rtol
      )
  
imports: import unittest
  import numpy
  import tensorflow
  import tensorflow_addons
  from tinygrad.tensor import Tensor
  from tinygrad.nn.optim import LAMB
label_desc: Use image node_id 2 for calling the ExternalTestOptim obj's underlying member method code with example usage: obj._test_lamb(steps, opts, atol, rtol) without return types