Revert : Skip testNonMatchingVariableCreation.

PiperOrigin-RevId: 301933594
Change-Id: Ifa25900442f291d25ef997fba09e61eb5aa1aafb
This commit is contained in:
Ran Chen 2020-03-19 17:59:17 -07:00 committed by TensorFlower Gardener
parent 30701c7344
commit 2260719374
1 changed files with 1 additions and 2 deletions

View File

@ -459,7 +459,6 @@ class MirroredVariableCreationTest(test.TestCase):
aggregation="invalid") aggregation="invalid")
def testNonMatchingVariableCreation(self, distribution): def testNonMatchingVariableCreation(self, distribution):
self.skipTest("b/123075960")
def model_fn(name): def model_fn(name):
v = variable_scope.variable(1.0, name=name) v = variable_scope.variable(1.0, name=name)
@ -467,7 +466,7 @@ class MirroredVariableCreationTest(test.TestCase):
return v return v
with distribution.scope(): with distribution.scope():
names = values.DistributedValues(("foo", "bar")) names = values.PerReplica(("foo", "bar"))
with self.assertRaises(RuntimeError): with self.assertRaises(RuntimeError):
_ = distribution.extended.call_for_each_replica(model_fn, args=(names,)) _ = distribution.extended.call_for_each_replica(model_fn, args=(names,))