AHj4ddlmZddlZddlmZmZddlmZddlm Z m Z eje 5ddl mZddderddlmZddlmZmZd d Zddd d  dd Zy#1swY0xYw)) annotationsN) TYPE_CHECKINGAny)dataframe_to_pydf)wrap_dfwrap_s)PySeries) DataFrame)SchemaDefinition SchemaDictc,tfddDS)z?Check if object looks like it supports the PyCapsule interface.c3JK|]}tt|dyw)N)callablegetattr).0attrobjs O/root/tools/cai/cai_env/lib/python3.12/site-packages/polars/_utils/pycapsule.py zis_pycapsule..s'  dD)*s #)__arrow_c_stream____arrow_c_array__)any)rs`r is_pycapsulers ? F)schemaschema_overridesrechunkct|drAd}ttj|}|j |j |}n^t|drAd}ttj |}|j |j |}nd|d}t||r|j}|s|rtt|||}|S)z&Convert PyCapsule object to DataFrame.rrz3object does not support PyCapsule interface; found  )rr) hasattrrr from_arrow_c_arrayto_frameunnestfrom_arrow_c_stream TypeErrorrrr)rrrr tmp_col_namesdfmsgs rpycapsule_to_framer+ss'( 8..s3 4 ZZ % , ,\ : * + 8//4 5 ZZ % , ,\ :CC7!Ln ZZ\ !  bBR S  Ir)rrreturnbool) rrrzSchemaDefinition | NonerzSchemaDict | Nonerr-r,r ) __future__r contextlibtypingrr$polars._utils.construction.dataframerpolars._utils.wraprrsuppress ImportError polars._plrr polarsr polars._typingr r rr+rrr9s"%B.Z%%$% ;'+*.   $(     !%%s A))A2