#include "module.h"

#include <ctranslate2/scoring.h>

#include "utils.h"

namespace ctranslate2 {

namespace python {

void register_scoring_result(py::module& m) {

py::class_<ScoringResult>(m, "ScoringResult", "A scoring result.")

.def_readonly("tokens", &ScoringResult::tokens,

"The scored tokens.")

.def_readonly("log_probs", &ScoringResult::tokens_score,

"Log probability of each token")

.def("__repr__", [](const ScoringResult& result) {

return "ScoringResult(tokens=" + std::string(py::repr(py::cast(result.tokens)))

+ ", log_probs=" + std::string(py::repr(py::cast(result.tokens_score)))

+ ")";

})

;

declare_async_wrapper<ScoringResult>(m, "AsyncScoringResult");

}

}

}