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")
|
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,))
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue