Hi, I'm trying to run the basic symbolic expressions example and I'm getting an error.
import jax
from jaxtyping import Float, Array
from typeguard import typechecked
@typechecked
def full(size: int, fill: float) -> Float[Array, "{size}"]:
return jax.numpy.full((size,), fill)
full(10, 1.0)
# AnnotationError: Cannot process symbolic axis '{size}' as some axis names have not been processed. In practice you should usually only use symbolic axes in annotations for return types, referring only to axes annotated for arguments.
Hi, I'm trying to run the basic symbolic expressions example and I'm getting an error.
jaxtyping 0.2.25, typeguard 2.13.3