def floating_named(
      cls,
      named_shape: Mapping[
          named_axes.AxisName | MultiDimVar,
          int | DimVar | RemainingAxisPlaceholder,
      ],
  ) -> ArraySpec:
    """Returns an `ArraySpec` with this named shape and `np.floating` dtype."""
    return cls(named_shape=named_shape, dtype=np.floating)
Returns an `ArraySpec` with this named shape and `np.floating` dtype.
penzai/core/shapecheck.py