diff --git a/SimPEG/Fields.py b/SimPEG/Fields.py index d9da317d..9baa32f8 100644 --- a/SimPEG/Fields.py +++ b/SimPEG/Fields.py @@ -235,11 +235,8 @@ class TimeFields(Fields): pointerFields = pointerFields.reshape(pointerShape, order='F') timeII = np.arange(self.survey.prob.nT + 1)[timeInd] - srcII = np.array(self.survey.srcList)[srcInd] - if isinstance(srcII, np.ndarray): - srcII = srcII.tolist() - if len(srcII) == 1: - srcII = srcII[0] + srcII = np.array(self.survey.srcList)[srcInd] + srcII = srcII.tolist() if timeII.size == 1: pointerShapeDeflated = self._correctShape(alias, ind, deflate=True) diff --git a/SimPEG/Tests/test_Fields.py b/SimPEG/Tests/test_Fields.py index bd75d712..629189b8 100644 --- a/SimPEG/Tests/test_Fields.py +++ b/SimPEG/Tests/test_Fields.py @@ -343,7 +343,7 @@ class FieldsTest_Time_Aliased(unittest.TestCase): count = [0] def alias(e, srcInd, timeInd): count[0] += 1 - self.assertTrue(srcInd is self.Src0) + self.assertTrue(srcInd[0] is self.Src0) return self.F.mesh.edgeCurl * e F = Problem.TimeFields(self.F.mesh, self.F.survey, knownFields={'e':'E'}, aliasFields={'b':['e','F',alias]}) e = np.random.rand(F.mesh.nE,1,nT)