Reference to FieldsTDEM

simpegEM/TDEM/BaseTDEM.py

@@ -1,6 +1,7 @@
 from SimPEG import Solver
 from SimPEG.Problem import BaseTimeProblem
 from simpegEM.Utils import Sources
+from SurveyTDEM import FieldsTDEM
 from scipy.constants import mu_0
 from SimPEG.Utils import sdiag, mkvc
 from SimPEG import Utils, Mesh

simpegEM/TDEM/TDEM_b.py

@@ -1,7 +1,7 @@
 from BaseTDEM import ProblemBaseTDEM
 from SimPEG.Utils import mkvc
 import numpy as np
-from SurveyTDEM import SurveyTDEM1D
+from SurveyTDEM import SurveyTDEM1D, FieldsTDEM
 
 class ProblemTDEM_b(ProblemBaseTDEM):
     """

simpegEM/TDEM/__init__.py

@@ -1,3 +1,3 @@
 from BaseTDEM import ProblemBaseTDEM
-from SurveyTDEM import SurveyTDEM1D
+from SurveyTDEM import SurveyTDEM1D, FieldsTDEM
 from TDEM_b import ProblemTDEM_b