From 08d759c2c339aca9e343015f8433317b1e73bbaf Mon Sep 17 00:00:00 2001 From: Zachary Garrett Date: Wed, 7 Aug 2024 10:05:40 -0700 Subject: [PATCH] Internal change. PiperOrigin-RevId: 660433633 --- .../python/core/backends/mapreduce/form_utils_test.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/tensorflow_federated/python/core/backends/mapreduce/form_utils_test.py b/tensorflow_federated/python/core/backends/mapreduce/form_utils_test.py index 9b0da8e258..335d302366 100644 --- a/tensorflow_federated/python/core/backends/mapreduce/form_utils_test.py +++ b/tensorflow_federated/python/core/backends/mapreduce/form_utils_test.py @@ -25,7 +25,7 @@ from tensorflow_federated.python.core.backends.mapreduce import forms from tensorflow_federated.python.core.backends.mapreduce import intrinsics as mapreduce_intrinsics from tensorflow_federated.python.core.backends.mapreduce import mapreduce_test_utils -from tensorflow_federated.python.core.backends.test import execution_contexts +from tensorflow_federated.python.core.backends.test import cpp_execution_contexts from tensorflow_federated.python.core.environments.tensorflow_frontend import tensorflow_computation from tensorflow_federated.python.core.impl.compiler import building_blocks from tensorflow_federated.python.core.impl.compiler import transformation_utils @@ -1505,5 +1505,7 @@ def test_binding_multiple_args_results_in_unique_names(self): if __name__ == '__main__': # The test execution context replaces all secure intrinsics with insecure # reductions. - execution_contexts.set_sync_test_cpp_execution_context(default_num_clients=1) + cpp_execution_contexts.set_sync_test_cpp_execution_context( + default_num_clients=1 + ) absltest.main()