From aa8f5a687e3d6997467ff506044fd5c344df462b Mon Sep 17 00:00:00 2001
From: rowanc1 <rowan@visiblegeology.com>
Date: Fri, 25 Apr 2014 15:47:17 -0700
Subject: [PATCH] model builder returns as vector.

---
 SimPEG/Utils/ModelBuilder.py | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/SimPEG/Utils/ModelBuilder.py b/SimPEG/Utils/ModelBuilder.py
index 462b9cfd..f17dac89 100644
--- a/SimPEG/Utils/ModelBuilder.py
+++ b/SimPEG/Utils/ModelBuilder.py
@@ -1,6 +1,7 @@
 import numpy as np
 import scipy.ndimage as ndi
 import scipy.sparse as sp
+from matutils import mkvc
 
 
 def getIndecesBlock(p0,p1,ccMesh):
@@ -81,7 +82,7 @@ def defineBlock(ccMesh,p0,p1,vals=[0,1]):
 
     sigma[ind] = vals[0]
 
-    return sigma
+    return mkvc(sigma)
 
 def defineElipse(ccMesh, center=[0,0,0], anisotropy=[1,1,1], slope=10., theta=0.):
     G = ccMesh.copy()
@@ -135,7 +136,7 @@ def defineTwoLayers(ccMesh,depth,vals=[0,1]):
 
     sigma[ind] = vals[0];
 
-    return sigma
+    return mkvc(sigma)
 
 def scalarConductivity(ccMesh,pFunction):
     """
@@ -150,7 +151,7 @@ def scalarConductivity(ccMesh,pFunction):
 
     sigma = pFunction(*CC)
 
-    return sigma
+    return mkvc(sigma)