Make primary property on DistributedValue private.

PiperOrigin-RevId: 295829087 Change-Id: I5c0bf035780f92de197578c979301ac900dc7afc
hixio-mh · Feb 18, 2020 · b6152b7 · b6152b7
1 parent 4bbbcbc
commit b6152b7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sonnet/src/distribute/replicator_test.py b/sonnet/src/distribute/replicator_test.py
@@ -73,7 +73,7 @@ def test_variable_synchronization_default(self, replicator_fn, create_var):
       self.skipTest("No replicator supplied.")
     v = create_var(replicator)
     self.assertEqual(tf.VariableSynchronization.ON_READ,
-                     v.primary.synchronization)
+                     v.values[0].synchronization)
 
   @test_utils.combined_named_parameters(replicator_utils.named_replicators(),
                                         all_variable_creators())