From 625c89f6102738a0e7b1c67e7d9810badfa45f00 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Chlo=C3=A9=20Kiddon?= <loeki@google.com>
Date: Wed, 5 Feb 2025 06:51:01 -0800
Subject: [PATCH] Remove `tff.federated_computation`.

Use `federated_language.federated_computation` instead.

PiperOrigin-RevId: 723492580
---
 .github/workflows/publish.yaml   | 1 -
 RELEASE.md                       | 1 +
 examples/simple_fedavg/README.md | 3 ++-
 tensorflow_federated/BUILD       | 1 -
 tensorflow_federated/__init__.py | 1 -
 5 files changed, 3 insertions(+), 4 deletions(-)

diff --git a/.github/workflows/publish.yaml b/.github/workflows/publish.yaml
index d9999000df..79a28fed63 100644
--- 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
index 5f83e33727..6730a8b198 100644
--- 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
index ed86fe77a4..5034d74981 100644
--- 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
index 55641191a9..28e1ae6512 100644
--- 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
index 7f5fac54c1..3754b6ddf4 100644
--- 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):