#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)))
            + ")";
        })
        ;
    }

  }
}