diff --git a/MaxText/configs/base.yml b/MaxText/configs/base.yml index 2fc059923..cf9ce4259 100644 --- 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 index 42e532f72..582389017 100644 --- 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)