Method _predict_single

lit_nlp/components/static_preds.py:73–78

(self, example: JsonDict)

    }

  def _predict_single(self, example: JsonDict):
    key = self.key_fn(example)
    if key not in self._index:
      raise KeyError(
          f'Example {key} not found in stored predictions: {str(example)}')
    return self._index[key]

  ##
  # LIT API implementation