Bug Fix: pass the SoB_only_matrix.shape when creating SoB_matrix_spar…

…se in the 4D KDE w/o set gamma case
icecube · Feb 7, 2025 · cab4e77 · cab4e77
1 parent 25437c0
commit cab4e77
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/flarestack/core/llh.py b/flarestack/core/llh.py
@@ -1526,7 +1526,7 @@ def joint_SoB(dataset, gamma):
 
                 SoB_matrix_sparse = sparse.csr_matrix(
                     (data, SoB_only_matrix.indices, SoB_only_matrix.indptr),
-                    shape=SoB_only_matrix,
+                    shape=SoB_only_matrix.shape,
                 )
 
                 SoB_sum = SoB_matrix_sparse.sum(axis=0)