Skip to content

Commit 63d4ba6

Browse files
sagipecopybara-github
authored andcommitted
Update equinox version
PiperOrigin-RevId: 822666606
1 parent 541219b commit 63d4ba6

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

requirements-jax.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,6 @@ optax>=0.2.3
77
chex>=0.1.87
88
tfp-nightly[jax]
99
# equinox 0.11.8 and above produces an error with HyperVolumeScalarization.
10-
equinox==0.11.7
10+
equinox
1111
jaxtyping>=0.2.34
1212
typeguard<=2.13.3

vizier/_src/algorithms/designers/scalarization.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -114,7 +114,7 @@ def __call__(
114114
class LinearAugmentedScalarization(Scalarization):
115115
"""Scalarization augmented with a linear sum.
116116
117-
See https://arxiv.org/pdf/1904.05760.pdf.
117+
See https://arxiv.org/pdf/1904.05760.pdf
118118
"""
119119

120120
weights: jt.Float[jax.Array, '*Num #Obj'] = eqx.field(converter=jnp.asarray)

0 commit comments

Comments
 (0)