Internal Changes.

PiperOrigin-RevId: 742424475
Change-Id: Ie4d5654a02d39ed1e27ad849467732c86cfceb3a

Author: liyuying0000

fleetbench/parallel/weights.py

@@ -65,7 +65,7 @@ def _ParseBenchmarkWeights(
 
 
 def GetDCTaxWeights() -> dict[str, float]:
-  """Read the weights for each benchmark from the data/weights.csv file.
+  """Read the weights for each benchmark from the corresponding weights.csv file.
 
   Returns:
       A dictionary containing the weights for each benchmark.