diff --git a/SimPEG/Utils/__init__.py b/SimPEG/Utils/__init__.py index 250146c1..18c1994f 100644 --- a/SimPEG/Utils/__init__.py +++ b/SimPEG/Utils/__init__.py @@ -1,6 +1,6 @@ from matutils import * from codeutils import * -from meshutils import exampleLrmGrid, meshTensor, closestPoints, readUBCTensorMesh, writeUBCTensorMesh, writeUBCTensorModel, readVTRFile, writeVTRFile +from meshutils import * from curvutils import volTetra, faceInfo, indexCube from interputils import interpmat from CounterUtils import * diff --git a/SimPEG/Utils/codeutils.py b/SimPEG/Utils/codeutils.py index 4a9a28a7..bfd00889 100644 --- a/SimPEG/Utils/codeutils.py +++ b/SimPEG/Utils/codeutils.py @@ -17,7 +17,7 @@ def memProfileWrapper(towrap, *funNames): For example:: - foo_mem = memProfile(foo,'my_func') + foo_mem = memProfileWrapper(foo,['my_func']) fooi = foo_mem() for i in range(5): fooi.my_func()