code: def output_spec(self):
      ret = self.base.output_spec()
      token_gradient_keys = utils.find_spec_keys(
          ret, lit_types.TokenGradients
      )
      for k in token_gradient_keys:
          ret.pop(k, None)
      return ret
  
imports: from typing import Optional
  from absl.testing import absltest
  from absl.testing import parameterized
  from lit_nlp.api import dataset
  from lit_nlp.api import model
  from lit_nlp.api import types
  from lit_nlp.components import hotflip
  from lit_nlp.lib import utils
  import numpy
label_desc: Use image node_id 3 for calling the ModelWithoutGradients obj's underlying member method code with example usage: obj.output_spec() and returns: ret