Fixed log to show how many slices are available during start up

lukebaumann · lukebaumann · commit c046863b4df3 · 2025-05-30T22:28:06.000Z
diff --git a/MaxText/elastic_train.py b/MaxText/elastic_train.py
@@ -438,7 +438,7 @@ def wait_for_all_slices(
   good_slice_indices = elastic_manager.get_slice_availability()
   while len(good_slice_indices) < elastic_manager.total_slice_count:
     max_logging.log(
-        f"Only {elastic_manager.good_slice_count} slices out of {elastic_manager.total_slice_count} available. "
+        f"Only {len(good_slice_indices)} slices out of {elastic_manager.total_slice_count} available. "
         f"Sleeping for {wait_period} seconds."
     )
     time.sleep(wait_period)

Original file line number	Diff line number	Diff line change
`@@ -438,7 +438,7 @@ def wait_for_all_slices(`
`438`	`438`	`good_slice_indices = elastic_manager.get_slice_availability()`
`439`	`439`	`while len(good_slice_indices) < elastic_manager.total_slice_count:`
`440`	`440`	`max_logging.log(`
`441`		`- f"Only {elastic_manager.good_slice_count} slices out of {elastic_manager.total_slice_count} available. "`
	`441`	`+ f"Only {len(good_slice_indices)} slices out of {elastic_manager.total_slice_count} available. "`
`442`	`442`	`f"Sleeping for {wait_period} seconds."`
`443`	`443`	`)`
`444`	`444`	`time.sleep(wait_period)`