i!UddlmZddlmZddlmZmZmZmZm Z m Z ddl m Z erddl mZmZddlmZddlmZddlmZGd d eZGd d eZeeeeefZd ed<GddeZGdde ZGdde ZGdde ZGdde Z Gdde Z!Gdde Z"GddZ#Gdd e$Z%Gd!d"Z&e&jOde&_(e&jOd#e&_)y$)%) annotations)IntEnum) TYPE_CHECKINGAnyClassVarLiteralProtocol TypedDict)issue_unstable_warning)IterableSequence) TypeAlias) PolarsBuffer) PolarsColumnc0eZdZdZdZdZdZdZdZdZ dZ d Z y ) DlpackDeviceTypez3Integer enum for device type codes matching DLPack. N) __name__ __module__ __qualname____doc__CPUCUDA CPU_PINNEDOPENCLVULKANMETALVPIROCM]/var/www/pod-logistic/pod-ai/venv/lib/python3.12/site-packages/polars/interchange/protocol.pyrrs-= C DJ F F E C Dr(rc,eZdZdZdZdZdZdZdZdZ dZ y ) DtypeKinda Integer enum for data types. Attributes ---------- INT : int Matches to signed integer data type. UINT : int Matches to unsigned integer data type. FLOAT : int Matches to floating point data type. BOOL : int Matches to boolean data type. STRING : int Matches to string data type (UTF-8 encoded). DATETIME : int Matches to datetime data type. CATEGORICAL : int Matches to categorical data type. rrrN) rrrrINTUINTFLOATBOOLSTRINGDATETIME CATEGORICALr'r(r)r+r+$s,* C D E D FHKr(r+rDtypec$eZdZdZdZdZdZdZdZy)ColumnNullTypea Integer enum for null type representation. Attributes ---------- NON_NULLABLE : int Non-nullable column. USE_NAN : int Use explicit float NaN value. USE_SENTINEL : int Sentinel value besides NaN. USE_BITMASK : int The bit is set/unset representing a null on a certain position. USE_BYTEMASK : int The byte is set/unset representing a null on a certain position. rrrrrN) rrrr NON_NULLABLEUSE_NAN USE_SENTINEL USE_BITMASK USE_BYTEMASKr'r(r)r9r9Fs""LGLKLr(r9c0eZdZUdZded<ded<ded<y) ColumnBufferszBuffers backing a column.ztuple[PolarsBuffer, Dtype]dataz!tuple[PolarsBuffer, Dtype] | NonevalidityoffsetsNrrrr__annotations__r'r(r)r@r@_s# %$ 0/ /.r(r@c0eZdZUdZded<ded<ded<y) CategoricalDescriptionz$Description of a categorical column.bool is_orderedz Literal[True] is_dictionaryr categoriesNrDr'r(r)rGrGrs.  r(rGcDeZdZdZeddZeddZddZd dZy) BufferzInterchange buffer object.cy)zBuffer size in bytes.Nr'selfs r)bufsizezBuffer.bufsizer(cy)z-Pointer to start of the buffer as an integer.Nr'rOs r)ptrz Buffer.ptrrRr(cy)z-Represent this structure as DLPack interface.Nr'rOs r) __dlpack__zBuffer.__dlpack__rRr(cy)zCDevice type and device ID for where the data in the buffer resides.Nr'rOs r)__dlpack_device__zBuffer.__dlpack_device__rRr(Nreturnint)rZr)rZz#tuple[DlpackDeviceType, int | None]) rrrrpropertyrQrTrVrXr'r(r)rMrM~s6$ $$<<<Rr(rMceZdZdZd dZed dZeddZeddZeddZ eddZ eddZ d d Z ddd Z dd Zy )ColumnzInterchange column object.cy)zSize of the column in elements.Nr'rOs r)sizez Column.sizerRr(cy)zOOffset of the first element with respect to the start of the underlying buffer.Nr'rOs r)offsetz Column.offsetrRr(cy)zData type of the column.Nr'rOs r)dtypez Column.dtyperRr(cy)z7Description of the categorical data type of the column.Nr'rOs r)describe_categoricalzColumn.describe_categoricalrRr(cy)z7Description of the null representation the column uses.Nr'rOs r) describe_nullzColumn.describe_nullrRr(cy)z"Number of null elements, if known.Nr'rOs r) null_countzColumn.null_countrRr(cy)zThe metadata for the column.Nr'rOs r)metadatazColumn.metadatarRr(cy)z3Return the number of chunks the column consists of.Nr'rOs r) num_chunkszColumn.num_chunksrRr(Ncy)z.Return an iterator yielding the column chunks.Nr'rPn_chunkss r) get_chunkszColumn.get_chunksrRr(cy)z6Return a dictionary containing the underlying buffers.Nr'rOs r) get_bufferszColumn.get_buffersrRr(rY)rZr7)rZrG)rZztuple[ColumnNullType, Any]rZ int | NonerZzdict[str, Any]N)rqrvrZIterable[Column])rZr@)rrrrr`r\rbrdrfrhrjrlrnrrrtr'r(r)r^r^s$.^^''FFFF11++B=Er(r^ceZdZUdZded< d ddZeddZddZddZ ddZ dd Z dd Z dd Z dd Zdd ZddZdddZy) DataFramezInterchange dataframe object.z ClassVar[int]versioncyzNConvert to a dataframe object implementing the dataframe interchange protocol.Nr'rP nan_as_null allow_copys r) __dataframe__zDataFrame.__dataframe__rRr(cy)zThe metadata for the dataframe.Nr'rOs r)rlzDataFrame.metadatarRr(cy)z.Return the number of columns in the dataframe.Nr'rOs r) num_columnszDataFrame.num_columnsrRr(cy)z9Return the number of rows in the dataframe, if available.Nr'rOs r)num_rowszDataFrame.num_rowsrRr(cy)z7Return the number of chunks the dataframe consists of..Nr'rOs r)rnzDataFrame.num_chunksrRr(cy)zReturn the column names.Nr'rOs r) column_nameszDataFrame.column_namesrRr(cy)z,Return the column at the indicated position.Nr')rPis r) get_columnzDataFrame.get_columnrRr(cy)z&Return the column with the given name.Nr')rPnames r)get_column_by_namezDataFrame.get_column_by_namerRr(cy)z(Return an iterator yielding the columns.Nr'rOs r) get_columnszDataFrame.get_columnsrRr(cy)zACreate a new dataframe by selecting a subset of columns by index.Nr')rPindicess r)select_columnszDataFrame.select_columnsrRr(cy)z@Create a new dataframe by selecting a subset of columns by name.Nr')rPnamess r)select_columns_by_namez DataFrame.select_columns_by_namerRr(Ncy)z8Return an iterator yielding the chunks of the dataframe.Nr'rps r)rrzDataFrame.get_chunksrRr(FT)rrHrrHrZr{rwrYru)rZz Iterable[str])rr[rZr^)rstrrZr^)rZry)rz Sequence[int]rZr{)rz Sequence[str]rZr{rx)rqrvrZzIterable[DataFrame])rrrrrErr\rlrrrnrrrrrrrrr'r(r)r{r{s' "]]]  ]..=HF';57POGr(r{c(eZdZdZ d ddZy)SupportsInterchangezHDataframe that supports conversion into an interchange dataframe object.cyr~r'rs r)rz!SupportsInterchange.__dataframe__rRr(Nr)rrHrrHrZr)rrrrrr'r(r)rrs2R"]]]  ]r(rc eZdZdZdZdZdZdZy) Endiannessz.Enum indicating the byte-order of a data type.<>=|N)rrrrLITTLEBIGNATIVENAr'r(r)rrs8 F C F Br(rceZdZdZy)CopyNotAllowedErrorzMException raised when a copy is required, but `allow_copy` is set to `False`.N)rrrrr'r(r)rrsWr(rcteZdZUdZded<d dZed dZed dZed dZ ed dZ dd Z y ) CompatLevelz#Data structure compatibility level.r[_versioncd}t|)Nz0it is not allowed to create a CompatLevel object) TypeError)rPmsgs r)__init__zCompatLevel.__init__s@nr(cFtjt}||_|Srx)r__new__r)r| compat_levels r) _with_versionzCompatLevel._with_versions "**;7 ' r(c"tjSrx)r_future1r'r(r)_newestzCompatLevel._newests###r(c@tdtjS)z Get the highest supported compatibility level. .. warning:: Highest compatibility level is considered **unstable**. It may be changed at any point without it being considered a breaking change. z=using the highest compatibility level is considered unstable.)r rrr'r(r)newestzCompatLevel.newests   K ""$$r(c"tjS)zGet the most compatible level.)r _compatibler'r(r)oldestzCompatLevel.oldest s&&&r(c~d|jjd|jjd|jdS)Nr.z: r) __class__rrrrOs r)__repr__zCompatLevel.__repr__%s84>>,,-Qt~~/J/J.K2dmm_\]^^r(N)rZNone)r|r[rZr)rZr)rZr) rrrrrEr staticmethodrrrrrr'r(r)rrse-M $$ % %''_r(rrN)* __future__renumrtypingrrrrr r polars._utils.unstabler collections.abcr r rpolars.interchange.bufferrpolars.interchange.columnrrr+tupler[rr7rEr9r@rGrMr^r{rr RuntimeErrorrrrrrr'r(r)rs":2 66 w >Cc12y2W2/I/& Y RXR$%EX%EP,G,G^](]  X,X'_'_T&33A6 "003 r(