ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-02-07 21:33:25 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix matplotlib binding support

Browse Source
This commit is contained in:
allegroai 2020-01-10 13:35:07 +02:00
parent f65ef3e757
commit 70624f469b
1 changed files with 9 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
trains/binding/import_bind.py
View File

@ -46,17 +46,23 @@ class PostImportHookPatching(object):
@staticmethod
def _patched_import3(name, globals=None, locals=None, fromlist=(), level=0):
base_name = name.split('.')[0]
already_imported = (not base_name) or (base_name in sys.modules)
name_parts = name.split('.')
base_name = name_parts[0]
second_name = '.'.join(name_parts[:2]) if len(name_parts) > 1 else None
base_already_imported = (not base_name) or (base_name in sys.modules)
second_already_imported = (not second_name) or (second_name in sys.modules)
mod = builtins.__org_import__(
name,
globals=globals,
locals=locals,
fromlist=fromlist,
level=level)
if not already_imported and base_name in PostImportHookPatching._post_import_hooks:
if not base_already_imported and base_name in PostImportHookPatching._post_import_hooks:
for hook in PostImportHookPatching._post_import_hooks[base_name]:
hook()
if not second_already_imported and second_name in PostImportHookPatching._post_import_hooks:
for hook in PostImportHookPatching._post_import_hooks[second_name]:
hook()
return mod
@staticmethod
@ -70,4 +76,3 @@ class PostImportHookPatching(object):
def remove_on_import(name, func):
if name in PostImportHookPatching._post_import_hooks and func in PostImportHookPatching._post_import_hooks[name]:
PostImportHookPatching._post_import_hooks[name].remove(func)