From 624ceef0a6a2dd982dee826465d46805475c6733 Mon Sep 17 00:00:00 2001 From: Michael Reneer Date: Wed, 26 Feb 2025 12:19:17 -0800 Subject: [PATCH] Remove deprecated `ComputationBuildingBlock.proto` method. PiperOrigin-RevId: 731410182 --- .../compiled_computation_transformations.py | 4 ++-- .../tensorflow_building_block_factory_test.py | 12 ++++++------ 2 files changed, 8 insertions(+), 8 deletions(-) diff --git a/tensorflow_federated/python/core/environments/tensorflow_backend/compiled_computation_transformations.py b/tensorflow_federated/python/core/environments/tensorflow_backend/compiled_computation_transformations.py index 2a6fe059e0..b398b52f43 100644 --- a/tensorflow_federated/python/core/environments/tensorflow_backend/compiled_computation_transformations.py +++ b/tensorflow_federated/python/core/environments/tensorflow_backend/compiled_computation_transformations.py @@ -77,7 +77,7 @@ def optimize_tensorflow_comp(tf_computation, config_proto): py_typecheck.check_type( tf_computation, federated_language.framework.CompiledComputation ) - tf_proto = tf_computation.proto + tf_proto = tf_computation.to_proto() graph_spec_obj = _unpack_proto_into_graph_spec(tf_proto) optimized_graph_spec = graph_optimizations.optimize_graph_spec( @@ -159,7 +159,7 @@ def transform(self, comp): comp, federated_language.framework.CompiledComputation ) new_comp_proto = tensorflow_computation_transformations.disable_grappler_for_partitioned_calls( - comp.proto + comp.to_proto() ) return ( federated_language.framework.CompiledComputation( diff --git a/tensorflow_federated/python/core/environments/tensorflow_backend/tensorflow_building_block_factory_test.py b/tensorflow_federated/python/core/environments/tensorflow_backend/tensorflow_building_block_factory_test.py index 46b05a18b7..2c09a28e68 100644 --- a/tensorflow_federated/python/core/environments/tensorflow_backend/tensorflow_building_block_factory_test.py +++ b/tensorflow_federated/python/core/environments/tensorflow_backend/tensorflow_building_block_factory_test.py @@ -41,7 +41,7 @@ def test_constructs_tensor_zero(self): self.assertTrue( np.array_equal( tensorflow_computation_test_utils.run_tensorflow( - tensor_zero.function.proto + tensor_zero.function.to_proto() ), np.zeros([2, 2]), ) @@ -56,7 +56,7 @@ def test_create_unnamed_tuple_zero(self): self.assertEqual(tuple_zero.type_signature, tuple_type) self.assertIsInstance(tuple_zero, federated_language.framework.Call) result = tensorflow_computation_test_utils.run_tensorflow( - tuple_zero.function.proto + tuple_zero.function.to_proto() ) self.assertLen(result, 2) self.assertTrue(np.array_equal(result[0], np.zeros([2, 2]))) @@ -75,7 +75,7 @@ def test_create_named_tuple_one(self): self.assertEqual(tuple_zero.type_signature, tuple_type) self.assertIsInstance(tuple_zero, federated_language.framework.Call) result = tensorflow_computation_test_utils.run_tensorflow( - tuple_zero.function.proto + tuple_zero.function.to_proto() ) self.assertLen(result, 2) self.assertTrue(np.array_equal(result.a, np.ones([2, 2]))) @@ -104,7 +104,7 @@ def test_create_federated_tensor_one(self): self.assertTrue( np.array_equal( tensorflow_computation_test_utils.run_tensorflow( - fed_zero.argument.function.proto + fed_zero.argument.function.to_proto() ), np.ones([2, 2]), ) @@ -134,7 +134,7 @@ def test_create_federated_named_tuple_one(self): ) self.assertIsInstance(fed_zero.argument, federated_language.framework.Call) result = tensorflow_computation_test_utils.run_tensorflow( - fed_zero.argument.function.proto + fed_zero.argument.function.to_proto() ) self.assertLen(result, 2) self.assertTrue(np.array_equal(result.a, np.ones([2, 2]))) @@ -165,7 +165,7 @@ def test_create_named_tuple_of_federated_tensors_zero(self): ) self.assertIsInstance(fed_zero.argument, federated_language.framework.Call) actual_result = tensorflow_computation_test_utils.run_tensorflow( - fed_zero.argument.function.proto + fed_zero.argument.function.to_proto() ) self.assertTrue(np.array_equal(actual_result, np.zeros([2, 2])))