
    j4                        d dl mZ d dlZd dlmZmZ d dlmZ d dlm	Z	m
Z
  ej        e          5  d dlmZ ddd           n# 1 swxY w Y   erd dlmZ d dlmZmZ ddZddddddZdS )    )annotationsN)TYPE_CHECKINGAny)dataframe_to_pydf)wrap_dfwrap_s)PySeries)	DataFrame)SchemaDefinition
SchemaDictobjr   returnboolc                :     t           fddD                       S )z?Check if object looks like it supports the PyCapsule interface.c              3  T   K   | ]"}t          t          |d                     V  #d S )N)callablegetattr).0attrr   s     \/home/longshao/multi-rider-rag/.venv/lib/python3.11/site-packages/polars/_utils/pycapsule.py	<genexpr>zis_pycapsule.<locals>.<genexpr>   sK         	dD))**         )__arrow_c_stream____arrow_c_array__)any)r   s   `r   is_pycapsuler      s;        ?     r   F)schemaschema_overridesrechunkr   SchemaDefinition | Noner   SchemaDict | Noner   r
   c                  t          | d          rLd}t          t          j        |                     }|                    |                              |          }nqt          | d          rLd}t          t          j        |                     }|                    |                              |          }nd| d}t          |          |r|                                }|s|rt          t          |||                    }|S )z&Convert PyCapsule object to DataFrame.r    r   z3object does not support PyCapsule interface; found  )r   r   )hasattrr   r	   from_arrow_c_arrayto_frameunnestfrom_arrow_c_stream	TypeErrorr   r   r   )r   r   r   r   tmp_col_namesdfmsgs           r   pycapsule_to_framer/      s    s'((  8.s3344ZZ%%,,\::	*	+	+  8/4455ZZ%%,,\::LCLLLnn ZZ\\ 
! 
bBRSSS
 
 Ir   )r   r   r   r   )
r   r   r   r    r   r!   r   r   r   r
   )
__future__r   
contextlibtypingr   r   $polars._utils.construction.dataframer   polars._utils.wrapr   r   suppressImportErrorpolars._plrr	   polarsr
   polars._typingr   r   r   r/    r   r   <module>r;      sZ   " " " " " "     % % % % % % % % B B B B B B . . . . . . . .Z%% % %$$$$$$% % % % % % % % % % % % % % %  <      ;;;;;;;;    '+*.       s   AAA