Fixed bugs

This commit is contained in:
Gudni Karl
2015-02-19 19:13:05 -08:00
parent 1549ced52d
commit fbf4370a78
2 changed files with 4 additions and 4 deletions
+1 -1
View File
@@ -31,7 +31,7 @@ for freq in np.logspace(3,-1,5):
# Survey MT
survey = simpegmt.SurveyMT.SurveyMT(srcList)
## Setup the problem object
## Setup the problem objec
problem = simpegmt.ProblemMT.MTProblem(M)
problem.pair(survey)
+3 -3
View File
@@ -1,4 +1,4 @@
from SimPEG import Survey, Utils, Problem, np, sp
from SimPEG import Survey, Utils, Problem, np, sp, mkvc
from scipy.constants import mu_0
class RxMT(Survey.BaseRx):
@@ -259,7 +259,7 @@ class DataMT(Survey.Data):
dt = dtRI
for uniFL in uniFLmarr:
mTemp = rec2ndarr(mArrRec[np.ma.where(mArrRec[['freq','x','y','z']].data == np.array(uniFL))][impList]).sum(axis=0)
dataBlock = simpeg.mkvc(np.concatenate((rec2ndarr(uniFL),mTemp.data)),2).T
dataBlock = mkvc(np.concatenate((rec2ndarr(uniFL),mTemp.data)),2).T
try:
outArr = np.concatenate((outArr,dataBlock),axis=0)
except NameError as e:
@@ -271,7 +271,7 @@ class DataMT(Survey.Data):
for uniFL in uniFLmarr:
mTemp = simpeg.mkvc(rec2ndarr(mArrRec[np.ma.where(mArrRec[['freq','x','y','z']].data == np.array(uniFL))][impList]).sum(axis=0),2).T
compBlock = np.sum(mTemp.data.reshape((4,2))*np.array([[1,1j],[1,1j],[1,1j],[1,1j]]),axis=1).copy().view(dt[4::])
dataBlock = simpeg.mkvc(recFunc.merge_arrays((np.array(uniFL),compBlock),flatten=True),2).T
dataBlock = mkvc(recFunc.merge_arrays((np.array(uniFL),compBlock),flatten=True),2).T
try:
outArr = recFunc.stack_arrays((outArr,dataBlock),usemask=False)
except NameError as e: