Method output_spec

lit_nlp/lib/testing_utils.py:49–50

(self)

    return self._input_spec

  def output_spec(self):
    return {'scores': lit_types.RegressionScore(parent='label')}

  def predict_minibatch(self, inputs: list[JsonDict], **kw):
    return self.predict(inputs)