compactified code for non trivial vectors

Author: juangmendoza19

pygsti/modelmembers/povms/__init__.py

@@ -580,10 +580,8 @@ def calc_physical_subspace(dense_ideal_povm, epsilon = 1e-4):
                     _,S,Vt = _np.linalg.svd(J)
 
                     #Only return nontrivial singular vectors
-                    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
+                    Vt = Vt[S > 1e-13, :].reshape((-1, Vt.shape[1]))
+                    return Vt
 
 
                 phys_directions = calc_physical_subspace(dense_ideal_povm)

pygsti/modelmembers/states/__init__.py

@@ -289,10 +289,8 @@ def calc_physical_subspace(ideal_prep, epsilon = 1e-4):
                         _,S,Vt = _np.linalg.svd(J)
 
                         #Only return nontrivial singular vectors
-                        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
+                        Vt = Vt[S > 1e-13, :].reshape((-1, Vt.shape[1]))
+                        return Vt
 
                     phys_directions = calc_physical_subspace(dense_state)