pub(crate) fn py_joint_psth( py: Python<'_>, train_a: PyReadonlyArray1<'_, i32>, train_b: PyReadonlyArray1<'_, i32>, bin_size: usize, ) -> Py<PyArray2<f64>>