pub(crate) fn py_mutual_information( train_a: PyReadonlyArray1<'_, i32>, train_b: PyReadonlyArray1<'_, i32>, bin_size: usize, ) -> f64