diff --git a/src/kbmod/filters/sigma_g_filter.py b/src/kbmod/filters/sigma_g_filter.py index 6ab42d48..87cdd67e 100644 --- a/src/kbmod/filters/sigma_g_filter.py +++ b/src/kbmod/filters/sigma_g_filter.py @@ -182,7 +182,7 @@ def compute_clipped_sigma_g_matrix(self, lh): A N x T matrix of Booleans indicating if each point is valid (True) or has been filtered (False). """ - inds_valid = self.sigma_g_jax_fn(jnp.asarray(lh)) + inds_valid = self.sigma_g_jax_fn(jnp.asarray(lh)).block_until_ready() return inds_valid