change enable_monitoring

google · Oct 2, 2023 · a5767b4 · a5767b4
1 parent 3e374e8
commit a5767b4
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/MaxText/configs/base.yml b/MaxText/configs/base.yml
@@ -177,6 +177,6 @@ stack_trace_interval_seconds: 600  # Stack trace collection frequency in seconds
 use_iota_embed: False
 
 #Monitoring parameters - Export in-workload metrics to Cloud monitoring
-enable_cloud_monitoring: True
+enable_cloud_monitoring: False
 cloud_monitoring_dashboard: "https://pantheon.corp.google.com/monitoring/dashboards?project="
 cloud_zone: "" # zone name for cloud jobs - used for cloud metrics emitting
diff --git a/MaxText/tests/cloud_monitoring_test.py b/MaxText/tests/cloud_monitoring_test.py
@@ -30,6 +30,7 @@ def test_write_time_series_step(self):
     pyconfig.initialize(sys.argv + ['configs/base.yml'], per_device_batch_size=1, run_name='test', mesh_axes = ['data'],
                         logical_axis_rules = [['batch', 'data']],
                         data_sharding = ['data'],
+                        enable_cloud_monitoring=True,
                         cloud_zone='us-central2-b')
     monitoring_api.create_custom_metric('test_metric', "This is an example metric")
     create_time_series_result = monitoring_api.write_time_series_step('test_metric', True, pyconfig, 1)