Update sigma_g_filter.py

dirac-institute · Dec 13, 2024 · 189ebb4 · 189ebb4
1 parent 034f307
commit 189ebb4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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