z iSrSSKJr SSKJr SSKJrJrJr SSK r SSK J r J r \\S4r\"SS \55r"S S \5rSS jrSS jrSSjrg)z+ Array shape related classes and functions ) annotations)Iterable)ProtocolUnionruntime_checkableN) ArrayLikeNDArrayzIterable[ShapeInput]cT\rSrSrSr\SSj5r\S Sj5r\S Sj5rSr g) ShapedaProtocol that defines what it means to be a shaped object. Note that static type checkers will classify ``numpy.ndarray`` as being :class:`Shaped`. Moreover, since this protocol is runtime-checkable, we will even have ``isinstance(, Shaped) == True``. c[S5e)zThe array shape of this object.z7A `Shaped` protocol must implement the `shape` propertyNotImplementedErrorselfs \/home/james-whalen/.local/lib/python3.13/site-packages/qiskit/primitives/containers/shape.pyshape Shaped.shape%s""[\\c[S5e)z.The number of array dimensions of this object.z6A `Shaped` protocol must implement the `ndim` propertyrrs rndim Shaped.ndim*""Z[[rc[S5e)zVThe total dimension of this object, i.e. the product of the entries of :attr:`~shape`.z6A `Shaped` protocol must implement the `size` propertyrrs rsize Shaped.size/rrN)returntuple[int, ...])rint) __name__ __module__ __qualname____firstlineno____doc__propertyrrr__static_attributes__rrrr r sI]]\\\\rr cZ\rSrSr%SrS\S'Sr\S5r\S5r \S5r S r g ) ShapedMixin5zXMixin class to create :class:`~Shaped` types by only providing :attr:`_shape` attribute.r_shapecL[U5RSURS3$)Nz())typer!rrs r__repr__ShapedMixin.__repr__:s$t*%%&b B77rcUR$N)r+rs rrShapedMixin.shape=s {{rc,[UR5$r1)lenr+rs rrShapedMixin.ndimAs4;;rcZ[[R"UR[S95$)N)dtype)r npprodr+rs rrShapedMixin.sizeEs2774;;c233rrN) r!r"r#r$r%__annotations__r.r&rrrr'rrrr)r)5sLb 8  44rr)c\[U[5(aU$[R"U5$)zCoerce the input into an object with a shape attribute. Copies are avoided. Args: arr: The object to coerce. Returns: Something that is :class:`~Shaped`, and always ``numpy.ndarray`` if the input is not already :class:`~Shaped`. ) isinstancer r8asarray)arrs r array_coercer@Js$#v ::c?rc## UHYn[U[5(a[U5ShvN M.[U5U:Xa[U5v ML[ SUS35e gN6![ [ 4an[ SUS35UeSnAff=f7f)z Yield one integer at a time. Args: arg: Integers or iterables of integers, possibly nested, to be yielded. Yields: The provided integers in depth-first recursive order. Raises: ValueError: If an input is not an iterable or an integer. Nz Expected z to be iterable or an integer.)r=r_flatten_to_intsr ValueError TypeErrorRecursionError)argitemexs rrBrB[s W$))+D111Td"$i 9TF2P!QRR2 >* Wy.LMNTV V WsNB #A%A#A%B A%B A% B #A%%B 5BB  B c*[[U55$)a Flatten the input into a single tuple of integers, preserving order. Args: shapes: Integers or iterables of integers, possibly nested. Returns: A tuple of integers. Raises: ValueError: If some member of ``shapes`` is not an integer or iterable. )tuplerB)shapess r shape_tuplerLts !&) **r)r?zArrayLike | ShapedrzNDArray | Shaped)rF ShapeInputrz Iterable[int])rKrMrr)r% __future__rcollections.abcrtypingrrrnumpyr8 numpy.typingrr r rMr r)r@rBrLrrrrSsk#$55+ 3.. / 4\X\\04&4*"W2 +r