experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add __reduce__ method in virtual pip root due to lazy loading

Browse Source 
Fixes pickling issue #32159

Tested manually: just applied patch to tensorflow nightly and checked required imports.

PiperOrigin-RevId: 276301497
Change-Id: I6b3b6ae8b1218b43c31403ea7cc595ed11136ff9
This commit is contained in:
Mihai Maruseac 2019-10-23 10:26:19 -07:00 committed by TensorFlower Gardener
parent f11328078f
commit 353b8a1adc
2 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tensorflow/virtual_root_template_v1.__init__.py
View File

@ -54,6 +54,9 @@ class _LazyLoader(_types.ModuleType):
module = self._load()
return dir(module)
def __reduce__(self):
return __import__, (self.__name__,)
# Forwarding a module is as simple as lazy loading the module from the new path
# and then registering it to sys.modules using the old path

3
tensorflow/virtual_root_template_v2.__init__.py
View File

@ -54,6 +54,9 @@ class _LazyLoader(_types.ModuleType):
module = self._load()
return dir(module)
def __reduce__(self):
return __import__, (self.__name__,)
# Forwarding a module is as simple as lazy loading the module from the new path
# and then registering it to sys.modules using the old path