From b0fe0dc8667b257d980472e087a9cc64857297b8 Mon Sep 17 00:00:00 2001
From: hengtaoguo
 <hengtaoguo@t1v-n-3bc70025-w-0.us-central2-b.c.tpu-prod-env-multipod.internal>
Date: Thu, 25 Jul 2024 23:07:54 +0000
Subject: [PATCH] Fix maxtext profiler failures

---
 MaxText/tests/profiler_test.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/MaxText/tests/profiler_test.py b/MaxText/tests/profiler_test.py
index 1c2cf3780e..c4ef6b59ef 100644
--- a/MaxText/tests/profiler_test.py
+++ b/MaxText/tests/profiler_test.py
@@ -66,8 +66,8 @@ def test_device_trace_contains_threads(self):
       if "name" in event and event["name"] == "thread_name":
         thread_names.append((event["args"]["name"]))
     expected_threads = [
-        "TensorFlow Name Scope",
-        "TensorFlow Ops",
+        "Framework Name Scope",
+        "Framework Ops",
         "XLA Modules",
         "XLA Ops",
         "XLA TraceMe",