
    LpjV0                       % S SK Jr  S SKJr  S SKJrJrJrJr  S SK	J
r
JrJrJrJrJrJrJrJrJrJr  S SKJrJrJr  S SKJr  \(       aI  S SKJrJrJrJrJ r J!r!  S SKJ"r"  S S	K#J$r$  S S
K%J&r&  S SKJ'r'J(r(  S SK)J*r*  S SK+J,r,J-r-J.r.J/r/J0r0J1r1J2r2  \r3S\4S'   / SQr5 " S S\\\
4   5      r6 " S S\\\4   5      r7 " S S\6\\4   \\\4   5      r8 " S S\6\\4   \\\\4   5      r9 " S S\8\\4   \\\\\\4   5      r:g)    )annotations)partial)TYPE_CHECKINGAnyProtocoloverload)CompliantExprTCompliantExprT_coCompliantFrameTCompliantLazyFrameTDepthTrackingExprTEagerDataFrameT
EagerExprTEagerSeriesT_co	LazyExprTNativeFrameTNativeSeriesT)exclude_column_namesget_column_namespassthrough_column_names)is_numpy_array_2d)
CollectionIterableIteratorKeysViewMappingSequence)	TypeAlias)TypeIs)CompliantSelectorNamespace)ImplementationVersion)DType)ConcatMethodCorrelationMethodInto1DArray	IntoDTypeNonNestedLiteralNormalizedPath_2DArrayr   
Incomplete)CompliantNamespaceDepthTrackingNamespaceEagerNamespaceLazyNamespacec                     \ rS rSr% S\S'   S\S'   \S#S j5       rS$S jrS%S jrS%S	 jr	S&S
 jr
        S'S jr        S(S jrS$S jrS)S jr      S*S jr      S*S jrS+S jrS+S jrS+S jrS+S jr      S,S jr        S-S jrS+S jr\S.S j5       rS+S jrS+S jrSS.       S/S jjrS0S jrS1S jrS2S  jrS!rg")3r,   >   r!   _implementationr"   _versionc                    g N selfs    Y/var/www/html/pdf-tiff/venv/lib/python3.13/site-packages/narwhals/_compliant/namespace.py_exprCompliantNamespace._exprC   s    -0    c                >    U R                   R                  [        U S9$ Ncontextr:   from_column_namesr   r7   s    r9   allCompliantNamespace.allF       zz++,<d+KKr<   c                H    U R                   R                  [        U5      U S9$ r>   r:   rB   r   r8   namess     r9   colCompliantNamespace.colI   #    zz++,DU,KUY+ZZr<   c                N    U R                   R                  [        [        US9U S9$ N)rI   r?   r:   rB   r   r   rH   s     r9   excludeCompliantNamespace.excludeL   +    zz++(6 , 
 	
r<   c                :    U R                   R                  " USU 06$ )Nr@   )r:   from_column_indices)r8   indicess     r9   nthCompliantNamespace.nthQ   s    zz--wEEEr<   c                   g r5   r6   )r8   abmethods       r9   corrCompliantNamespace.corrT       r<   c                   g r5   r6   )r8   rY   rZ   ddofs       r9   covCompliantNamespace.covW   r^   r<   c                    g r5   r6   r7   s    r9   lenCompliantNamespace.lenZ   s    Sr<   c                    g r5   r6   r8   valuedtypes      r9   litCompliantNamespace.lit[   s    WZr<   c                   g r5   r6   r8   ignore_nullsexprss      r9   all_horizontal!CompliantNamespace.all_horizontal\   r^   r<   c                   g r5   r6   rm   s      r9   any_horizontal!CompliantNamespace.any_horizontal_   r^   r<   c                    g r5   r6   r8   ro   s     r9   sum_horizontal!CompliantNamespace.sum_horizontalb       r<   c                    g r5   r6   rv   s     r9   mean_horizontal"CompliantNamespace.mean_horizontalc   s    r<   c                    g r5   r6   rv   s     r9   min_horizontal!CompliantNamespace.min_horizontald   ry   r<   c                    g r5   r6   rv   s     r9   max_horizontal!CompliantNamespace.max_horizontale   ry   r<   c                   g r5   r6   )r8   itemshows      r9   concatCompliantNamespace.concatf       r<   c                   g r5   r6   )r8   	separatorrn   ro   s       r9   
concat_strCompliantNamespace.concat_stri   r^   r<   c                    g r5   r6   rv   s     r9   listCompliantNamespace.listl   s    cr<   c                    g r5   r6   r7   s    r9   	selectorsCompliantNamespace.selectorsm   s    ADr<   c                    g r5   r6   rv   s     r9   structCompliantNamespace.structo   s    r<   c                    g r5   r6   rv   s     r9   coalesceCompliantNamespace.coalescep   s    #r<   ,r   c                   g r5   r6   r8   sourcer   kwdss       r9   scan_csvCompliantNamespace.scan_csvq   r   r<   c                    g r5   r6   r8   r   r   s      r9   scan_parquetCompliantNamespace.scan_parquett       TWr<   c                   g r5   r6   r8   datas     r9   from_nativeCompliantNamespace.from_nativev   s    r<   c                   g)z6Return `True` if `obj` can be passed to `from_native`.Nr6   r8   objs     r9   	is_nativeCompliantNamespace.is_nativew   s    r<   r6   N)returnztype[CompliantExprT])r   r	   )rI   strr   r	   )rU   zSequence[int]r   r	   )rY   r	   rZ   r	   r[   r%   r   r	   )rY   r	   rZ   r	   r`   intr   r	   )rh   r(   ri   IntoDType | Noner   r	   )ro   r	   rn   boolr   r	   )ro   r	   r   r	   )r   Iterable[CompliantFrameT]r   r$   r   r   )ro   r	   r   r   rn   r   r   r	   )r   z$CompliantSelectorNamespace[Any, Any])r   r)   r   r   r   r   r   r   )r   r)   r   r   r   r   )r   r   r   r   )r   r   r   zTypeIs[Any]) __name__
__module____qualname____firstlineno____annotations__propertyr:   rC   rJ   rP   rV   r\   ra   rd   rj   rp   rs   rw   r{   r~   r   r   r   r   r   r   r   r   r   r   r   __static_attributes__r6   r<   r9   r,   r,   >   s`   ##0 0L[

F$2?P	$2=@	 )Z$48	$48	 LLKK.8D	$14DH	 BD DCE:=$47GJ	 X3r<   r,   c                  Z    \ rS rSrSr      SS jr    S	S jr        S
S jrSrg)AlignDiagonal|   z2Mixin to help support `"diagonal*"` concatenation.c                    g r5   r6   rg   s      r9   rj   AlignDiagonal.lit   s    r<   c               2  ^ U Vs/ s H  o"R                  5       PM     nn[        U5      n[        [        U5      5      nUR	                  5       mU H,  nUR                  U4S jUR                  5        5       5        M.     U R                  XU5      $ s  snf )a  Prepare frames with differing schemas for vertical concatenation.

Adapted from [`convert_diagonal_concat`].

[`convert_diagonal_concat`]: https://github.com/pola-rs/polars/blob/c2412600210a21143835c9dfcb0a9182f462b619/crates/polars-plan/src/plans/conversion/dsl_to_ir/concat.rs#L10-L68
c              3  >   >#    U  H  u  pUT;  d  M  X4v   M     g 7fr5   r6   ).0nmri   seens      r9   	<genexpr>/AlignDiagonal.align_diagonal.<locals>.<genexpr>   s     WnRV"ns   
)collect_schemaiterdictnextkeysupdater   _align_diagonal)r8   framesframeschemas
it_schemasunionschemar   s          @r9   align_diagonalAlignDiagonal.align_diagonal   s     8>>ve'')v> ']
T*%&zz| FLLWfllnWW !##FU;; ?s   Bc           	        ^ ^^^^	 TR                  5       m	0 mSUU U4S jjmSUU	4S jjn[        XSS9 VVs/ s H  u  pVU" XVR                  5       5      PM     snn$ s  snnf )Nc              3     >#    U  HF  nTR                  U5      =nc+  TU   nTR                  S U5      R                  U5      =nTU'   Uv   MH     g 7fr5   )getrj   alias)missingnameexprri   
null_exprsr8   union_schemas       r9   iter_missing_exprs9AlignDiagonal._align_diagonal.<locals>.iter_missing_exprs   sX     &NN400D9(.E.2hhtU.C.I.I$.OOD:d+
	  s   AAc                d   > TU-
  =n(       a  U R                   " T" U5      6 n U R                  " T6 $ r5   )with_columnssimple_select)dfcolumnsr   r   union_namess      r9   align,AlignDiagonal._align_diagonal.<locals>.align   s8    %//w/__&8&AB##[11r<   F)strict)r   zIterable[str]r   zIterator[CompliantExprT_co])r   r   r   zKeysView[str]r   r   )r   zip)
r8   r   r   r   r   r   r   r   r   r   s
   `  `   @@@r9   r   AlignDiagonal._align_diagonal   sg     #'')35
	 		2 	2 "%VU!C
!C %'!C
 	
 
s   "Ar6   N)rh   r(   ri   r   r   r
   )r   zCollection[CompliantFrameT]r   Sequence[CompliantFrameT])r   r   r   zIterable[Mapping[str, DType]]r   zMapping[str, DType]r   r   )	r   r   r   r   __doc__rj   r   r   r   r6   r<   r9   r   r   |   sb    < % .> 	 <1<	"<&
)
 /
 *	

 
#
r<   r   c                  2    \ rS rSrSS jrSS jrSS jrSrg)	r-      c                >    U R                   R                  [        U S9$ r>   rA   r7   s    r9   rC   DepthTrackingNamespace.all   rE   r<   c                H    U R                   R                  [        U5      U S9$ r>   rG   rH   s     r9   rJ   DepthTrackingNamespace.col   rL   r<   c                N    U R                   R                  [        [        US9U S9$ rN   rO   rH   s     r9   rP   DepthTrackingNamespace.exclude   rR   r<   r6   N)r   r   )rI   r   r   r   )r   r   r   r   rC   rJ   rP   r   r6   r<   r9   r-   r-      s    L[
r<   r-   c                  P    \ rS rSr\SS j5       r\S	S j5       rS
S jrSS jrSr	g)r/      c                6    U R                   R                  5       $ r5   r2   _backend_versionr7   s    r9   r   LazyNamespace._backend_version       ##4466r<   c                    g r5   r6   r7   s    r9   
_lazyframeLazyNamespace._lazyframe   s    7:r<   c               8    U R                   R                  U5      $ r5   )r   
_is_nativer   s     r9   r   LazyNamespace.is_native   s    ))#..r<   c                   U R                   R                  U5      (       a  U R                   R                  XS9$ S[        U5      R                  < 3n[        U5      eNr?   zUnsupported type: )r   r  r   typer   	TypeErrorr8   r   msgs      r9   r   LazyNamespace.from_native   sP    ??%%d++??..t.BB"4:#6#6"9:nr<   r6   Nr   ztuple[int, ...])r   ztype[CompliantLazyFrameT])r   r   r   zTypeIs[NativeFrameT])r   zNativeFrameT | Anyr   r   )
r   r   r   r   r   r   r   r   r   r   r6   r<   r9   r/   r/      s0     7 7 : :/r<   r/   c                     \ rS rSr\SS j5       r\SS j5       r\SS j5       rSS.       SS jjrSS jr	SS.       SS	 jjr
SS
 jr S       S S jjr S       S!S jjrS"S jr\S#S j5       r\S$S j5       r    S%S jr\S&S'S jj5       r\      S(S j5       r S     S)S jjrS*S jr    S+S jrS*S jr      S,S jrSrg)-r.      c                6    U R                   R                  5       $ r5   r   r7   s    r9   r   EagerNamespace._backend_version   r   r<   c                    g r5   r6   r7   s    r9   
_dataframeEagerNamespace._dataframe   s    36r<   c                    g r5   r6   r7   s    r9   _seriesEagerNamespace._series   s    03r<   r   r   c                   g r5   r6   r   s       r9   read_csvEagerNamespace.read_csv   r   r<   c                    g r5   r6   r   s      r9   read_parquetEagerNamespace.read_parquet   r   r<   c               ,    U R                   " U4SU0UD6$ )Nr   )r  r   s       r9   r   EagerNamespace.scan_csv   s     }}VAyADAAr<   c                (    U R                   " U40 UD6$ r5   )r  r   s      r9   r   EagerNamespace.scan_parquet   s      0400r<   Nc                    g r5   r6   )r8   whenthen	otherwises       r9   _if_then_elseEagerNamespace._if_then_else   s    
 r<   c           	        ^ ^^^ SUUU U4S jjnT R                   R                  U[        TSS 5      [        TSS 5      TS9$ )Nc                  > U R                  T5      nUR                  nU R                  T	5      nTc1  U" X5      u  pTR                  UR                  UR                  5      nONU R                  T5      nU" XU5      u  pnTR                  UR                  UR                  UR                  5      nUR	                  U5      /$ r5   )_evaluate_single_output_expr_align_full_broadcastr%  native_with_native)
r   predicate_sr   then_sresultotherwise_sr$  	predicater8   r#  s
         r9   func&EagerNamespace.when_then.<locals>.func   s    99)DK55E44T:F &+K&@#++K,>,>N ==iH38k3Z0[++&&{7I7I ''/00r<   _evaluate_output_namesc                    S/$ )Nliteralr6   )_dfs    r9   <lambda>*EagerNamespace.when_then.<locals>.<lambda>  s    YKr<   _alias_output_names)r2  evaluate_output_namesalias_output_namesr@   )r   r   r   zSequence[EagerSeriesT_co])r:   _from_callablegetattr)r8   r1  r#  r$  r2  s   ```` r9   	when_thenEagerNamespace.when_then   sQ    	1 	1  zz((").0G#  't-BDI ) 
 	
r<   c               |    U R                   R                  U5      =(       d    U R                  R                  U5      $ r5   )r  r  r  r   s     r9   r   EagerNamespace.is_native  s+    ))#.N$,,2I2I#2NNr<   c                   g r5   r6   r   s     r9   r   EagerNamespace.from_native  s    EHr<   c                   g r5   r6   r   s     r9   r   rD    s    FIr<   c               .   U R                   R                  U5      (       a  U R                   R                  XS9$ U R                  R                  U5      (       a  U R                  R                  XS9$ S[	        U5      R
                  < 3n[        U5      er  )r  r  r   r  r  r   r  r	  s      r9   r   rD    s     ??%%d++??..t.BB<<""4((<<++D+??"4:#6#6"9:nr<   c                   g r5   r6   r8   r   r   s      r9   
from_numpyEagerNamespace.from_numpy"  s    WZr<   c                   g r5   r6   rH  s      r9   rI  rJ  %  s     r<   c                   [        U5      (       a  U R                  R                  XU S9$ U R                  R                  XS9$ )N)r   r@   r?   )r   r  rI  r  rH  s      r9   rI  rJ  *  sA     T""??--d4-PP||&&t&::r<   c                   g r5   r6   r8   dfss     r9   _concat_diagonalEagerNamespace._concat_diagonal4      PSr<   c                   g r5   r6   rN  s     r9   _concat_horizontal!EagerNamespace._concat_horizontal5  s    r<   c                   g r5   r6   rN  s     r9   _concat_verticalEagerNamespace._concat_vertical8  rR  r<   c                  U Vs/ s H  o3R                   PM     nnUS:X  a  U R                  U5      nO6US:X  a  U R                  U5      nOUS:X  a  U R                  U5      nO[        eU R
                  R                  XPS9$ s  snf )N
horizontalverticaldiagonalr?   )r+  rT  rW  rP  NotImplementedErrorr  r   )r8   r   r   itemrO  r+  s         r9   r   EagerNamespace.concat9  s     (--ut{{u-,,,S1FJ**3/FJ**3/F%%**6*@@ .s   Br6   r  )r   ztype[EagerDataFrameT])r   ztype[EagerSeriesT_co])r   r)   r   r   r   r   r   r   )r   r)   r   r   r   r   r5   )r"  r   r#  r   r$  zNativeSeriesT | Noner   r   )r1  r   r#  r   r$  zEagerExprT | Noner   r   )r   r   r   z$TypeIs[NativeFrameT | NativeSeriesT])r   r   r   r   )r   r   r   r   )r   z"NativeFrameT | NativeSeriesT | Anyr   !EagerDataFrameT | EagerSeriesT_co).)r   Noner   r&   r   r   )r   .Mapping[str, IntoDType] | Sequence[str] | Noner   r*   r   r   )r   rb  r   zInto1DArray | _2DArrayr   r`  )rO  zSequence[NativeFrameT]r   r   )rO  zSequence[NativeFrameT | Any]r   r   )r   zIterable[EagerDataFrameT]r   r$   r   r   )r   r   r   r   r   r   r  r  r  r  r   r   r%  r?  r   r   r   rI  rP  rT  rW  r   r   r6   r<   r9   r.   r.      s    7 7 6 63 3:=$47GJ	 X:=B$B47BGJB	B
1 +/	  (	
 
 W[
#
+5
BS
	
8O H HI I6	* Z Z)W	  BF	; ?	;$;
 
+; T/	 TA.A8DA	Ar<   r.   N);
__future__r   	functoolsr   typingr   r   r   r   narwhals._compliant.typingr	   r
   r   r   r   r   r   r   r   r   r   narwhals._utilsr   r   r   narwhals.dependenciesr   collections.abcr   r   r   r   r   r   r   typing_extensionsr   narwhals._compliant.selectorsr    r!   r"   narwhals.dtypesr#   narwhals.typingr$   r%   r&   r'   r(   r)   r*   r+   r   __all__r,   r   r-   r/   r.   r6   r<   r9   <module>ro     s   "  9 9    
 4  !(H7%    J	;/>"AB ;|4
H_.??@ 4
n
(::;_001
 *I56 )\9:(nA?J67_oz<VWnAr<   