Extend unvec() to accept 1d arrays too

rigetti · Jan 10, 2019 · a65c671 · a65c671
1 parent b4ee278
commit a65c671
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/forest_qcvv/superop_conversion.py b/forest_qcvv/superop_conversion.py
@@ -40,8 +40,8 @@ def unvec(vector):
     :param vector: A N^2 x 1 numpy array.
     :return: Returns a column vector with  N x N rows.
     """
-    row, col = vector.shape
-    dim = int(np.sqrt(row))
+    n_row = vector.shape[0]
+    dim = int(np.sqrt(n_row))
     matrix = vector.reshape(dim, dim).T
     return matrix