updates to survey and mkvc

2026-08-11 11:26:01 +08:00 · 2014-03-10 22:58:25 -07:00
parent fcaff5ed64
commit 6fdc7d4a62
2 changed files with 11 additions and 1 deletions
@@ -30,6 +30,9 @@ def mkvc(x, numDims=1):
    if type(x) == np.matrix:
        x = np.array(x)

+    if hasattr(x, 'tovec'):
+        x = x.tovec()
+
    assert type(x) == np.ndarray, "Vector must be a numpy array"

    if numDims == 1: