Changed the name space to correspond with FDEM.

simpegMT/BaseMT.py

@@ -1,8 +1,7 @@
 from SimPEG import SolverLU as SimpegSolver, PropMaps, Utils, mkvc, sp, np
 from SimPEG.EM.FDEM.FDEM import BaseFDEMProblem
-from SurveyMT import SurveyMT
-from DataMT import DataMT
-from FieldsMT import FieldsMT
+from SurveyMT import Survey, Data
+from FieldsMT import BaseMTFields
 
 
 class BaseMTProblem(BaseFDEMProblem):
@@ -11,37 +10,9 @@ class BaseMTProblem(BaseFDEMProblem):
         BaseFDEMProblem.__init__(self, mesh, **kwargs)
         Utils.setKwargs(self, **kwargs)
     # Set the default pairs of the problem
-    surveyPair = SurveyMT
-    dataPair = DataMT
-    fieldsPair = FieldsMT
-
-    # # Pickleing support methods
-    # def __getstate__(self):
-    #     '''
-    #     Method that makes the dictionary of the object pickleble, removes non-pickleble elements of the object.
-
-    #     Used when doing:
-    #         pickle.dump(pickleFile,object)
-    #     '''
-    #     odict = self.__dict__.copy()
-    #     # Remove fields that are not needed
-    #     del odict['hook']
-    #     del odict['setKwargs']
-    #     # Return the dict
-    #     return odict
-
-    # def __setstate__(self,odict):
-    #     '''
-    #     Function that sets a pickle dictionary in to an object.
-
-    #     Used when doing:
-    #         object = pickle.load(pickleFile)
-    #     '''
-    #     # Update the dict
-    #     self.__dict__.update(odict)
-    #     # Re-hook the methods to the object
-    #     Utils.codeutils.hook(self,Utils.codeutils.hook)
-    #     Utils.codeutils.hook(self,Utils.codeutils.setKwargs)
+    surveyPair = Survey
+    dataPair = Data
+    fieldsPair = BaseMTFields
 
     # Set the solver
     Solver = SimpegSolver