#include "module.h"
#include <ctranslate2/logging.h>
namespace ctranslate2 {
namespace python {
void register_logging(py::module& m) {
py::enum_<ctranslate2::LogLevel>(m, "LogLevel")
.value("Off", ctranslate2::LogLevel::Off)
.value("Critical", ctranslate2::LogLevel::Critical)
.value("Error", ctranslate2::LogLevel::Error)
.value("Warning", ctranslate2::LogLevel::Warning)
.value("Info", ctranslate2::LogLevel::Info)
.value("Debug", ctranslate2::LogLevel::Debug)
.value("Trace", ctranslate2::LogLevel::Trace)
.export_values();
m.def("set_log_level", &ctranslate2::set_log_level);
m.def("get_log_level", &ctranslate2::get_log_level);
}
}
}