ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-01-31 09:07:00 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix python 2.7 support

Browse Source
This commit is contained in:
allegroai 2019-07-06 23:03:14 +03:00
parent e40d4f2f41
commit a1cf210896
1 changed files with 10 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
trains/binding/absl_bind.py
View File

@ -1,4 +1,6 @@
""" absl-py FLAGS binding utility functions """
import six
from ..backend_interface.task.args import _Arguments
from ..config import running_remotely
@ -20,7 +22,10 @@ class PatchAbsl(object):
# noinspection PyBroadException
try:
from absl.flags import _defines
cls._original_DEFINE_flag = _defines.DEFINE_flag
if six.PY2:
cls._original_DEFINE_flag = staticmethod(_defines.DEFINE_flag)
else:
cls._original_DEFINE_flag = _defines.DEFINE_flag
_defines.DEFINE_flag = cls._patched_define_flag
except Exception:
# there is no absl
@ -28,7 +33,10 @@ class PatchAbsl(object):
try:
from absl.flags._flagvalues import FlagValues
cls._original_FLAGS_parse_call = FlagValues.__call__
if six.PY2:
cls._original_FLAGS_parse_call = staticmethod(FlagValues.__call__)
else:
cls._original_FLAGS_parse_call = FlagValues.__call__
FlagValues.__call__ = cls._patched_FLAGS_parse_call
except Exception:
# there is no absl