pub fn register(_py: Python<'_>, m: &PyModule) -> PyResult<()>
Register the classes to the python module