| 242 | } // namespace mlx |
| 243 | |
| 244 | PYBIND11_MODULE(_c, m) { |
| 245 | m.doc() = "mlx data"; |
| 246 | |
| 247 | py::enum_<TokenizeMode>(m, "TokenizeMode") |
| 248 | .value("Shortest", TokenizeMode::shortest) |
| 249 | .value("Rand", TokenizeMode::rand) |
| 250 | .export_values(); |
| 251 | |
| 252 | py::enum_<LoadAudioInfo>(m, "LoadAudioInfo") |
| 253 | .value("All", LoadAudioInfo::All) |
| 254 | .value("NumFrames", LoadAudioInfo::NumFrames) |
| 255 | .value("NumChannels", LoadAudioInfo::NumChannels) |
| 256 | .value("SampleRate", LoadAudioInfo::SampleRate) |
| 257 | .value("NumSeconds", LoadAudioInfo::NumSeconds) |
| 258 | .export_values(); |
| 259 | |
| 260 | init_mlx_data_stream(m); |
| 261 | init_mlx_data_buffer(m); |
| 262 | |
| 263 | py::module_ m_core = m.def_submodule("core", "mlx.data.core"); |
| 264 | init_mlx_data_core(m_core); |
| 265 | |
| 266 | m.attr("__version__") = TOSTRING(_VERSION_); |
| 267 | } |
nothing calls this directly
no test coverage detected