Skip to content

Commit 384b340

Browse files
committed
Merge pull request #398 from ferdnyc/no-distutils
Remove dependency on distutils.version
2 parents e2a5a5a + 59ae100 commit 384b340

1 file changed

Lines changed: 4 additions & 26 deletions

File tree

‎memory_profiler.py‎

Lines changed: 4 additions & 26 deletions
Original file line numberDiff line numberDiff line change
@@ -939,18 +939,9 @@ def mprun(self, parameter_s='', cell=None):
939939
from io import StringIO
940940
from memory_profiler import show_results, LineProfiler
941941

942-
# Local imports to avoid hard dependency.
943-
from distutils.version import LooseVersion
944-
import IPython
945-
ipython_version = LooseVersion(IPython.__version__)
946-
if ipython_version < '0.11':
947-
from IPython.genutils import page
948-
from IPython.ipstruct import Struct
949-
from IPython.ipapi import UsageError
950-
else:
951-
from IPython.core.page import page
952-
from IPython.utils.ipstruct import Struct
953-
from IPython.core.error import UsageError
942+
from IPython.core.page import page
943+
from IPython.utils.ipstruct import Struct
944+
from IPython.core.error import UsageError
954945

955946
# Escape quote markers.
956947
opts_def = Struct(T=[''], f=[])
@@ -1133,20 +1124,7 @@ def memit(self, line='', cell=None):
11331124

11341125
@classmethod
11351126
def register_magics(cls, ip):
1136-
from distutils.version import LooseVersion
1137-
import IPython
1138-
ipython_version = LooseVersion(IPython.__version__)
1139-
1140-
if ipython_version < '0.13':
1141-
try:
1142-
_register_magic = ip.define_magic
1143-
except AttributeError: # ipython 0.10
1144-
_register_magic = ip.expose_magic
1145-
1146-
_register_magic('mprun', cls.mprun.__func__)
1147-
_register_magic('memit', cls.memit.__func__)
1148-
else:
1149-
ip.register_magics(cls)
1127+
ip.register_magics(cls)
11501128

11511129

11521130
# commenting out due to failures with some versions of IPython

0 commit comments

Comments
 (0)