Method predict

lit_nlp/components/static_preds.py:91–105

Predict on known inputs. Args: inputs: input examples **kw: unused Returns: predictions Raises: KeyError if input not recognized

(self, inputs: Iterable[JsonDict], **kw)

    return list(self.predict(inputs))

  def predict(self, inputs: Iterable[JsonDict], **kw) -> Iterator[JsonDict]:
    """Predict on known inputs.

    Args:
      inputs: input examples
      **kw: unused

    Returns:
      predictions

    Raises:
      KeyError if input not recognized
    """
    # Implement predict() directly, since there's no need to batch.
    return map(self._predict_single, inputs)