Automated Code Change

PiperOrigin-RevId: 627964992
tensorflow · Apr 25, 2024 · f3c3844 · f3c3844
1 parent 49a561f
commit f3c3844
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/tensorflow/python/framework/BUILD b/tensorflow/python/framework/BUILD
@@ -3115,7 +3115,11 @@ tf_python_pybind_extension(
         "_errors_test_helper.pyi",
     ],
     deps = [
+        "//tensorflow/c:tf_status_headers",
+        "//tensorflow/core/platform:status",
         "//tensorflow/python/lib/core:pybind11_status",
+        "@com_google_absl//absl/status",
+        "@com_google_absl//absl/strings:cord",
         "@pybind11",
     ],
 )

diff --git a/tensorflow/python/framework/errors_test_helper.cc b/tensorflow/python/framework/errors_test_helper.cc
@@ -13,7 +13,11 @@ See the License for the specific language governing permissions and
 limitations under the License.
 ==============================================================================*/
 
+#include "absl/status/status.h"
+#include "absl/strings/cord.h"
 #include "pybind11/pybind11.h"  // from @pybind11
+#include "tensorflow/c/tf_status.h"
+#include "tensorflow/core/platform/status.h"
 #include "tensorflow/python/lib/core/pybind11_status.h"
 
 namespace tensorflow {