Merge pull request #3154 from AI-Hypercomputer:tyagiva-diagon-env

Google-ML-Automation · Google-ML-Automation · commit 597d16c3f4ed · 2026-02-18T12:38:22.000-08:00
PiperOrigin-RevId: 871995248
diff --git a/src/maxtext/common/managed_mldiagnostics.py b/src/maxtext/common/managed_mldiagnostics.py
@@ -74,5 +74,4 @@ def should_log_key(key, value):
         gcs_path=config.managed_mldiagnostics_dir,
         # TODO: b/455623960 - Remove the following once multi-region and prod support are enabled.
         region="us-central1",
-        environment="autopush",  # Default would be "prod" for formal launch.
     )

Original file line number	Diff line number	Diff line change
`@@ -74,5 +74,4 @@ def should_log_key(key, value):`
`74`	`74`	`gcs_path=config.managed_mldiagnostics_dir,`
`75`	`75`	`# TODO: b/455623960 - Remove the following once multi-region and prod support are enabled.`
`76`	`76`	`region="us-central1",`
`77`		`- environment="autopush", # Default would be "prod" for formal launch.`
`78`	`77`	`)`