Updated some comments in pyproject.toml

lukebaumann · lukebaumann · commit 01e26a27cffe · 2025-07-08T18:52:47.000Z
diff --git a/pyproject.toml b/pyproject.toml
@@ -110,11 +110,11 @@ tpu = [
     "axlearn[gcp]",
     "jax[tpu]==0.5.3",  # must be >=0.4.19 for compat with v5p.
 ]
-# For Pathways TPU training.
+# For TPU+Pathways single-controller training
 pathways-tpu = [
     "axlearn[gcp]",
     "jax==0.5.3",  # must be >=0.4.19 for compat with v5p.
-    "pathwaysutils @ git+https://github.com/AI-Hypercomputer/pathways-utils",  # For JAX+Pathways single-controller accelerator coordinator.
+    "pathwaysutils @ git+https://github.com/AI-Hypercomputer/pathways-utils",
 ]
 # Vertex AI tensorboard. TODO(markblee): Merge with `gcp`.
 vertexai_tensorboard = [

Original file line number	Diff line number	Diff line change
`@@ -110,11 +110,11 @@ tpu = [`
`110`	`110`	`"axlearn[gcp]",`
`111`	`111`	`"jax[tpu]==0.5.3", # must be >=0.4.19 for compat with v5p.`
`112`	`112`	`]`
`113`		`-# For Pathways TPU training.`
	`113`	`+# For TPU+Pathways single-controller training`
`114`	`114`	`pathways-tpu = [`
`115`	`115`	`"axlearn[gcp]",`
`116`	`116`	`"jax==0.5.3", # must be >=0.4.19 for compat with v5p.`
`117`		`- "pathwaysutils @ git+https://github.com/AI-Hypercomputer/pathways-utils", # For JAX+Pathways single-controller accelerator coordinator.`
	`117`	`+ "pathwaysutils @ git+https://github.com/AI-Hypercomputer/pathways-utils",`
`118`	`118`	`]`
`119`	`119`	# Vertex AI tensorboard. TODO(markblee): Merge with `gcp`.
`120`	`120`	`vertexai_tensorboard = [`