class_docstr: Input examples to a SingleEdgePredictor.
class_docstr_tok: ["Input","examples","to","a","SingleEdgePredictor","."]
code: def spec(self):
      return {
          "span1_embs": lit_types.Embeddings(),
          "span2_embs": lit_types.Embeddings(),
          "label": lit_types.Scalar(),
          "src_idx": lit_types.Scalar(),
          "edge_idx": lit_types.Scalar(),
      }
  
imports: import os
  from collections.abc import Iterable
  from typing import Optional
  from absl import logging
  from lit_nlp.api import dataset
  from lit_nlp.api import dtypes
  from lit_nlp.api import model
  from lit_nlp.api import types
  from lit_nlp.lib import utils
  import numpy
  import tensorflow
label_desc: Use image node_id 3 for calling the EdgeFeaturesDataset obj's underlying member method code with example usage: obj.spec() and returns: dict