Update test_profiler.py path in Maxtext tests

GoogleCloudPlatform · Jan 10, 2025 · 4477d8a · 4477d8a
1 parent 23bc11b
commit 4477d8a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dags/multipod/maxtext_profiling.py b/dags/multipod/maxtext_profiling.py
@@ -53,7 +53,7 @@
         "pip3 uninstall -y tensorflow",
         "pip3 install tf-nightly",
         "pip3 install tbp-nightly",
-        "python3 MaxText/tests/profiler_test.py",
+        "python3 end_to_end/test_profiler.py",
     )
     maxtext_v4_configs_test = gke_config.get_gke_config(
         time_out_in_min=60,