Remove tff.federated_computation.

Use `federated_language.federated_computation` instead. PiperOrigin-RevId: 723492580
JetBrains-Research · Feb 5, 2025 · 625c89f · 625c89f
1 parent a96a2d8
commit 625c89f
Show file tree

Hide file tree

Showing 5 changed files with 3 additions and 4 deletions.
diff --git a/.github/workflows/publish.yaml b/.github/workflows/publish.yaml
@@ -184,7 +184,6 @@ jobs:
           exit 1
         fi
         pip install --upgrade "${package}"
-        python -I -c "import tensorflow_federated as tff; print(tff.federated_computation(lambda: 'Hello World')())"
         python -I -c "import tensorflow_federated as tff; print(tff.__version__)"
 
   publish-package:

diff --git a/RELEASE.md b/RELEASE.md
@@ -253,6 +253,7 @@ and this project adheres to
 *   `tff.federated_aggregate`, use `federated_language.federated_aggregate`
     instead.
 *   `tff.federated_broadcast`, use `federated_language.federated_broadcast`
+*   `tff.federated_computation`, use `federated_language.federated_computation`
     instead.
 *   `tff.federated_eval`, use `federated_language.federated_eval` instead.
 *   `tff.federated_map`, use `federated_language.federated_map` instead.

diff --git a/examples/simple_fedavg/README.md b/examples/simple_fedavg/README.md
@@ -27,7 +27,8 @@ federated learning simulation.
 *   The orchestration logic binds together the local computations by wrapping
     them as `tff.tensorflow.computation`s and using key TFF functions like
     `federated_language.federated_broadcast` and
-    `federated_language.federated_map` inside a `tff.federated_computation`.
+    `federated_language.federated_map` inside a
+    `federated_language.federated_computation`.
 
 This EMNIST example can easily be adapted for experimental changes:
 

diff --git a/tensorflow_federated/BUILD b/tensorflow_federated/BUILD
@@ -24,7 +24,6 @@ py_library(
         "//tensorflow_federated/python/learning",
         "//tensorflow_federated/python/program",
         "//tensorflow_federated/python/simulation",
-        "@federated_language//federated_language",
     ],
 )
 

diff --git a/tensorflow_federated/__init__.py b/tensorflow_federated/__init__.py
@@ -28,7 +28,6 @@
 from tensorflow_federated.python.core.environments import jax
 from tensorflow_federated.python.core.environments import tensorflow
 from tensorflow_federated.version import __version__
-from federated_language import federated_computation
 # pylint: enable=g-bad-import-order,g-importing-member
 
 if sys.version_info < (3, 9):