commit | 9fe37b34f50a5334a70ff5cee88f8bede4ab5308 | [log] [tgz] |
---|---|---|
author | Chenkai Kuang <chenkai@google.com> | Wed Oct 21 12:45:04 2020 -0700 |
committer | TensorFlower Gardener <gardener@tensorflow.org> | Wed Oct 21 12:56:23 2020 -0700 |
tree | cf3ed03427a49576d00b6b221acb25e03cf9a1e4 | |
parent | 898c3b6d335cc8e50b53e69099d449b07326388e [diff] |
Enable tf.Module + sharded variable test given the integration is in place. PiperOrigin-RevId: 338319945 Change-Id: I6a2a0bd397f035e6fbce9fce4eda5e154ed38604
diff --git a/tensorflow/python/distribute/sharded_variable_test.py b/tensorflow/python/distribute/sharded_variable_test.py index 8c22dcc..2aa8cae 100644 --- a/tensorflow/python/distribute/sharded_variable_test.py +++ b/tensorflow/python/distribute/sharded_variable_test.py
@@ -364,7 +364,6 @@ self.assertAllEqual(variables, got) def test_tf_module(self): - self.skipTest('integration with tf.module is not added yet.') class Model(module.Module):