fixed problem with SV masking shape

Author: juangmendoza19

pygsti/modelmembers/povms/__init__.py

@@ -579,7 +579,7 @@ def calc_physical_subspace(dense_ideal_povm, epsilon = 1e-9):
 
                     _,S,Vt = _np.linalg.svd(J)
                     #Only return nontrivial singular vectors
-                    non_zero_mask = _np.abs(S) > 1e-13
+                    non_zero_mask = _np.where(_np.abs(S) > 1e-13)
                     non_trivial_vecs = Vt[non_zero_mask]
                     non_trivial_vecs = non_trivial_vecs.reshape(-1, Vt.shape[1])  # Reshape to ensure it's 2D
                     return non_trivial_vecs

pygsti/modelmembers/states/__init__.py

@@ -289,7 +289,7 @@ def calc_physical_subspace(ideal_prep, epsilon = 1e-9):
                         _,S,Vt = _np.linalg.svd(J)
 
                         #Only return nontrivial singular vectors
-                        non_zero_mask = _np.abs(S) > 1e-13
+                        non_zero_mask = _np.where(_np.abs(S) > 1e-13)
                         non_trivial_vecs = Vt[non_zero_mask]
                         non_trivial_vecs = non_trivial_vecs.reshape(-1, Vt.shape[1])  # Reshape to ensure it's 2D
                         return non_trivial_vecs