Revert : Skip testNonMatchingVariableCreation.
PiperOrigin-RevId: 301933594 Change-Id: Ifa25900442f291d25ef997fba09e61eb5aa1aafb
This commit is contained in:
parent
30701c7344
commit
2260719374
|
@ -459,7 +459,6 @@ class MirroredVariableCreationTest(test.TestCase):
|
|||
aggregation="invalid")
|
||||
|
||||
def testNonMatchingVariableCreation(self, distribution):
|
||||
self.skipTest("b/123075960")
|
||||
|
||||
def model_fn(name):
|
||||
v = variable_scope.variable(1.0, name=name)
|
||||
|
@ -467,7 +466,7 @@ class MirroredVariableCreationTest(test.TestCase):
|
|||
return v
|
||||
|
||||
with distribution.scope():
|
||||
names = values.DistributedValues(("foo", "bar"))
|
||||
names = values.PerReplica(("foo", "bar"))
|
||||
with self.assertRaises(RuntimeError):
|
||||
_ = distribution.extended.call_for_each_replica(model_fn, args=(names,))
|
||||
|
||||
|
|
Loading…
Reference in New Issue