g/ PdZddlmZddlmZddlZddlmZ dddZ dddZ dS)z* Object ND-array initialization function. ) annotations)SequenceN) ArrayLikeTarrrorder str | Nonecopybool list_typesSequence[type] | Nonereturn np.ndarrayc0t|tjr5|jtks||dur|t||}|St |t|}tj|t|}|s| |nsfd|t|}t||j krtdtj |tt||_|S) a;Convert an array-like of objects into an object array. .. note:: If the objects in the array like input define ``__array__`` methods this avoids calling them and will instead set the returned array values to the Python objects themselves. Args: arr: An array-like input. order: Optional, the order of the returned array (C, F, A, K). If None the default NumPy ordering of C is used. copy: If True make a copy of the input if it is already an array. list_types: Optional, a sequence of types to treat as lists of array element objects when inferring the array shape from the input. Returns: A NumPy ND-array with ``dtype=object``. Raises: ValueError: If the input cannot be coerced into an object array. NT)rr r )dtyperc4dkr|Sfd|DS)Nc8g|]}|dz D]}|S)rr).0sublistitem_flattenks _/granary/edev/quantum/lib/python3.11/site-packages/qiskit/primitives/containers/object_array.py z2object_array.._flatten..Hs9XXXxxQRUVQV?W?WXXtXXXXr)nestedrrs `rrzobject_array.._flattenDs/Avv XXXXXFXXXXrztInput object size does not match the inferred array shape. This most likely occurs when the input is a ragged array.)rcount) isinstancenpndarrayrobjectastype _infer_shapetupleemptyfilllensize ValueErrorfromiterflat)rrr r shapeobj_arr flattenedrs @r object_arrayr1s&8#rz"" 9  %"3tt||**V5t*<>W\ ) )M {9F#i..QQQ Nrobjtuple[type, ...]tuple[int, ...]ct|tjr|jSt|tg|RsdSt |}|dkr|fS|gt |d|RS)z7Infer the shape of an array-like object without castingrrr)r r!r"r.listr)r%)r2r r*s rr%r%Us}#rz""y cD.:.. / /r s88D qyyw  ?<A:>>> ? ??r)NTr) rrrr r r r r rr)r)r2rr r3rr4) __doc__ __future__rcollections.abcrnumpyr! numpy.typingrr1r%rrrr<s#"""""$$$$$$"""""" (* :::::z @ @ @ @ @ @ @r