code: def input_spec(self) -> dict[str, lit_types.LitType]:
      return {
          "sentence": lit_types.TextSegment(),
          "tokens_sentence": lit_types.Tokens(
              parent="sentence", required=False
          ),
          "input_embs_sentence": lit_types.TokenEmbeddings(
              align="tokens_sentence", required=False
          ),
      }
  
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 1 for calling the TestClassificationModel obj's underlying member method code with example usage: obj.input_spec() and returns: dict