Update src/stream_mapper/core/_core/base.py

Signed-off-by: Nathaniel Starkman <[email protected]>
GalOrrery · Feb 6, 2024 · 81233f6 · 81233f6
1 parent 3b22b02
commit 81233f6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/stream_mapper/core/_core/base.py b/src/stream_mapper/core/_core/base.py
@@ -333,7 +333,7 @@ def _ln_prior_coord_bnds(self, data: Data[Array], /) -> Array:
 
         where = reduce(
             self.xp.logical_or,
-            (~within_bounds(data[k], a[:, 0], b[:, 0]) for k, (a, b) in kab),
+            (~within_bounds(data[k], a, b) for k, (a, b) in kab),
             self.xp.zeros(shape, dtype=bool),
         )
         return self.xp.where(