From a31bbde597f56145e064136f92be70cb31d87329 Mon Sep 17 00:00:00 2001 From: Linh Tran Date: Thu, 10 Oct 2024 10:39:06 -0700 Subject: [PATCH] fix simulation bug --- src/generate_simulations.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/generate_simulations.py b/src/generate_simulations.py index 27adc3b..21ca926 100644 --- a/src/generate_simulations.py +++ b/src/generate_simulations.py @@ -79,7 +79,7 @@ def _calculate_probability(self, signature_matrix, exposure): """ time_points = [] - selected_signatures = signature_matrix[1:, np.where(np.intersect1d(signature_matrix[0, :], self._signatures))] + selected_signatures = signature_matrix[1:, np.where(np.isin(signature_matrix[0, :], self._signatures))[0]] selected_signatures = selected_signatures.reshape(selected_signatures.shape[0], len(self._signatures)).astype(float) for i in exposure.T: