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

Fix constructor of CommunicationOptions

Browse Source 
Fix #44768

PiperOrigin-RevId: 341888489
Change-Id: I625f2e5412a217a95e3306fc6345f378b932071b
This commit is contained in:
Ran Chen 2020-11-11 12:37:55 -08:00
parent ef82f4c66c
commit eff07b8226
2 changed files with 9 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tensorflow/python/distribute/collective_util.py
View File

@ -81,7 +81,11 @@ class _OptionsExported(object):
""" """
def __new__(cls, *args, **kwargs): def __new__(cls, *args, **kwargs):
return Options.__new__(Options, *args, **kwargs) # We expose a dummy class so that we can separate internal and public APIs.
# Note that __init__ won't be called on the returned object if it's a
# different class [1].
# [1] https://docs.python.org/3/reference/datamodel.html#object.__new__
return Options(*args, **kwargs)
def __init__(self, def __init__(self,
bytes_per_pack=0, bytes_per_pack=0,

5
tensorflow/python/distribute/collective_util_test.py
View File

@ -25,8 +25,11 @@ from tensorflow.python.eager import test
class OptionsTest(test.TestCase): class OptionsTest(test.TestCase):
def testCreateOptionsViaExportedAPI(self): def testCreateOptionsViaExportedAPI(self):
options = collective_util._OptionsExported() options = collective_util._OptionsExported(bytes_per_pack=1)
self.assertIsInstance(options, collective_util.Options) self.assertIsInstance(options, collective_util.Options)
self.assertEqual(options.bytes_per_pack, 1)
with self.assertRaises(ValueError):
collective_util._OptionsExported(bytes_per_pack=-1)
def testCreateOptionsViaHints(self): def testCreateOptionsViaHints(self):
with self.assertLogs() as cm: with self.assertLogs() as cm: