#include "module.h"

#include <ctranslate2/translator.h>

namespace ctranslate2 {
 namespace python {

 void register_translation_stats(py::module& m) {
 py::class_<ExecutionStats>(m, "ExecutionStats",
 "A structure containing some execution statistics.")

 .def_readonly("num_tokens", &ExecutionStats::num_tokens,
 "Number of output tokens.")
 .def_readonly("num_examples", &ExecutionStats::num_examples,
 "Number of processed examples.")
 .def_readonly("total_time_in_ms", &ExecutionStats::total_time_in_ms,
 "Total processing time in milliseconds.")

 .def("__repr__", [](const ExecutionStats& stats) {
 return "ExecutionStats(num_tokens=" + std::string(py::repr(py::cast(stats.num_tokens)))
 + ", num_examples=" + std::string(py::repr(py::cast(stats.num_examples)))
 + ", total_time_in_ms=" + std::string(py::repr(py::cast(stats.total_time_in_ms)))
 + ")";
 })
 ;
 }

 }
}