bi= d Z ddlZddlZddlZddlmZmZmZmZm Z m
Z
m Z m Z ddl
ZddlmZmZmZ ddlmZ G d de Z G d d
e Z G d d e Z e
d
eeef eed eeedf eedf e d
def d ZdedefdZ dede fdZ d!dededefdZ de eef deedf de ej@ dedef
d Z!dejD defd Z#dejD de fd Z$ded eddfd Z%y)"z+Helpers for interfacing array like objects. N)LiteralOptionalProtocolTupleType TypedDictUnioncast ) CNumericPtrDataType NumpyOrCupy) import_cupyc e Zd Zedd Zy)
_ArrayLikeArgc y N selfs N/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/xgboost/_data_utils.py__array_interface__z!_ArrayLikeArg.__array_interface__ s 14 NreturnArrayInf)__name__
__module__ __qualname__propertyr r r r r r s
4 4r r c e Zd Zedd Zy)_CudaArrayLikeArgc y r r r s r __cuda_array_interface__z*_CudaArrayLikeArg.__cuda_array_interface__ s 69r Nr )r r r r r$ r r r r" r" s
9 9r r" c < e Zd ZdZdefdZedeeef fd Z y)
TransformedDfz1Protocol class for storing transformed dataframe.r c y)z,Get a JSON-encoded list of array interfaces.Nr r s r array_interfacez TransformedDf.array_interface r c y)z"Return the shape of the dataframe.Nr r s r shapezTransformedDf.shape r) r N)
r r r __doc__bytesr( r r intr+ r r r r& r& s4 ;; ; 1uS#X 1 1r r& r .)datatypestrversionstridesr+ maskr0 r c ^ t | j d xr | j j S )z)Whether the numpy array has object dtype. hasobject)hasattrdtyper6 )r0 s r array_hasobjectr9 0 s" 4::{ + D
0D0DDr c t | r t d | j }d|v r|d j |d< t t j
| d }|S )z Make cuda array interface str..Arrayz3Wrapper type for communicating with numpy and cupy.N
_interfacer c | j S r )rH r s r r z7from_array_interface..Array.__array_interface__H s ?? "r rA c R t j | | _ t | j d | j d< | j d d | j d d f| j d< | j j dd }|t | | j d< y y )Nr+ r0 r r r3 )copyrH tupleget)r rA r3 s r r z7from_array_interface..Array.__array_interface__L s "ii 2DO',T__W-E'FDOOG $'*'*'DOOF # oo)))T:G"-27^ * #r c | j S r r r s r r$ z.Array.__cuda_array_interface__Y s ++ +r c || _ y r rO )r rA s r r$ z.Array.__cuda_array_interface__] s
'0D $r ) r r r r, rH r r __annotations__r r setterr$ r r r ArrayrG C s A)-
HX&- #(); #
#
# #
<
r? r@ s r r( r( s' $T*I$**Y/9M r fieldc \ d| j v r | j d t d| yy)z(Make sure no missing value in meta data.r4 Nz"Missing value is not allowed for: )r$ r= )r0 rk s r check_cudf_metarm s@ $/// ) )& 1 ==eWEFF
> 0r )F)&r, rK r` r> typingr r r r r r r r
numpyrV _typingr r
r compatr r r" r& r. boolstrr r9 r- rC rY numberrf ndarrayri r( rm r r r rv s} 1
S S S 7 7 5H 5
: :
1H 1 c4i 1:E#s(O,sCx j$
56
E( Et E
x E +H + ++ +\ {C c? ?
D"rzz "h ""** G+ GC GD Gr