Drop extra index

andersensam · andersensam · commit 5c7191d8411a · 2025-10-21T19:24:50.000Z
diff --git a/axlearn/common/optimizers.py b/axlearn/common/optimizers.py
@@ -39,7 +39,6 @@
 from jax import numpy as jnp
 from jax import sharding
 
-# from jax._src.sharding_impls import TransferToMemoryKind
 from optax._src import numerics
 
 from axlearn.common import flax_struct, schedule
diff --git a/pyproject.toml b/pyproject.toml
@@ -215,9 +215,6 @@ find-links = [
     "https://storage.googleapis.com/axlearn-wheels/wheels.html",
     "https://storage.googleapis.com/jax-releases/libtpu_releases.html",
 ]
-# extra-index-url = [
-#     "https://us-python.pkg.dev/ml-oss-artifacts-published/jax-public-nightly-artifacts-registry/simple"
-# ]
 no-build-isolation-package = [
     "apache-beam",
 ]

Original file line number	Diff line number	Diff line change
`@@ -215,9 +215,6 @@ find-links = [`
`215`	`215`	`"https://storage.googleapis.com/axlearn-wheels/wheels.html",`
`216`	`216`	`"https://storage.googleapis.com/jax-releases/libtpu_releases.html",`
`217`	`217`	`]`
`218`		`-# extra-index-url = [`
`219`		`-# "https://us-python.pkg.dev/ml-oss-artifacts-published/jax-public-nightly-artifacts-registry/simple"`
`220`		`-# ]`
`221`	`218`	`no-build-isolation-package = [`
`222`	`219`	`"apache-beam",`
`223`	`220`	`]`