§
    {Øjz?  ã                  óz  — d dl mZ d dlZd dlmZmZ d dlmZmZ d dlm	Z	m
Z
mZ 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 d d	lmZ d d
lmZmZmZ d dlmZm Z  d dl!m"Z" d dl#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z) e	r#d dl*m+Z+ d dl,mc m-Z. d dl#m/Z/ d dl0m1Z1m2Z2m3Z3m4Z4m5Z5  G d„ deeeej6        f         eeef         ¦  «        Z7dS )é    )ÚannotationsN)ÚdateÚdatetime)ÚpartialÚreduce)ÚTYPE_CHECKINGÚAnyÚcast)ÚDepthTrackingNamespaceÚLazyNamespace©ÚDaskLazyFrame©ÚDaskExpr)ÚDaskSelectorNamespace)Úalign_series_full_broadcastÚnarwhals_to_native_dtypeÚvalidate_comparand)Úcombine_alias_output_namesÚcombine_evaluate_output_names)Únative_to_narwhals_dtype)ÚImplementationÚensure_path_sourceÚis_nested_literalÚisinstance_or_issubclassÚnot_implementedÚvalidate_separators)ÚIterable)ÚVersion)ÚConcatMethodÚCorrelationMethodÚ	IntoDTypeÚNonNestedLiteralÚNormalizedSourcec                  ó*  — e Zd ZU ej        Zded<   ed>d„¦   «         Zed?d„¦   «         Z	ed@d	„¦   «         Z
dAd„ZddœdBd„ZdCd„ZdDd„ZdEd„ZdFd#„ZdFd$„ZdGd%„ZdHd*„ZdGd+„ZdGd,„ZdGd-„ZdId.„ZdGd/„Z	 dJdKd5„ZdLd:„ZdMd=„Z e¦   «         Z e¦   «         Zd0S )NÚDaskNamespacer   Ú_implementationÚreturnr   c                ó*   — t          j        | ¦  «        S ©N)r   Úfrom_namespace©Úselfs    ú�/Users/tedpeters/Documents/Projects/Claude Projects/Misc Tools/notename-sheets/.venv/lib/python3.11/site-packages/narwhals/_dask/namespace.pyÚ	selectorszDaskNamespace.selectors7   s   € å$Ô3°DÑ9Ô9Ð9ó    útype[DaskExpr]c                ó   — t           S r*   r   r,   s    r.   Ú_exprzDaskNamespace._expr;   s   € åˆr0   útype[DaskLazyFrame]c                ó   — t           S r*   r   r,   s    r.   Ú
_lazyframezDaskNamespace._lazyframe?   s   € åÐr0   Úversionr   ÚNonec               ó   — || _         d S r*   )Ú_version)r-   r7   s     r.   Ú__init__zDaskNamespace.__init__C   s   € ØˆŒˆˆr0   ú,)Ú	separatorÚsourcer$   r=   ÚstrÚkwdsr	   r   c               óª   — t          |d|¦  «         t          j        t          || j        ¦  «        fd|i|¤Ž}| j                             || ¬¦  «        S )N)ÚseprB   ©Úcontext)r   ÚddÚread_csvr   r'   r6   Úfrom_native)r-   r>   r=   r@   Únatives        r.   Úscan_csvzDaskNamespace.scan_csvF   si   € õ 	˜I x°Ñ6Ô6Ð6Ý”Ý˜v tÔ';Ñ<Ô<ð
ð 
ØBKð
ØOSð
ð 
ˆð Œ×*Ò*¨6¸4Ð*Ñ@Ô@Ð@r0   c                ó€   — | j                              t          j        t	          || j        ¦  «        fi |¤Ž| ¬¦  «        S )NrC   )r6   rG   rE   Úread_parquetr   r'   )r-   r>   r@   s      r.   Úscan_parquetzDaskNamespace.scan_parquetO   sH   € ØŒ×*Ò*ÝŒOÕ.¨v°tÔ7KÑLÔLÐUÐUÐPTÐUÐUØð +ñ 
ô 
ð 	
r0   Úvaluer#   ÚdtypeúIntoDType | Noner   c                ó   ‡ ‡‡— t          ‰¦  «        r&dt          ‰¦  «        j        › �}t          |¦  «        ‚t	          ‰‰ j        j        j        ¦  «        rd}t          |¦  «        ‚d
ˆˆ ˆfd„}‰                      |d„ d ‰ j        ¬	¦  «        S )Nz<Nested structures are not supported for Dask backend, found z3`List` literals are not supported for Dask backend.Údfr   r(   úlist[dx.Series]c                ó–  •— ‰�.t          ‰‰j        ¦  «        }t          j        ‰g|d¬¦  «        }n\t	          ‰t
          ¦  «        r0t	          ‰t          ¦  «        sd}t          j        ‰g|d¬¦  «        }nt          j        ‰gd¬¦  «        }| j        j        }t          j
        ||¬¦  «        }|d                              ¦   «         gS )NÚliteral)rN   Únamezdate32[pyarrow])rU   )Únpartitionsr   )r   r:   ÚpdÚSeriesÚ
isinstancer   r   Ú_native_framerV   rE   Úfrom_pandasÚ	to_series)rQ   Únative_dtypeÚnative_pd_seriesrV   Údask_seriesrN   r-   rM   s        €€€r.   ÚfunczDaskNamespace.lit.<locals>.func_   sÓ   ø€ ØÐ Ý7¸¸t¼}ÑMÔM�Ý#%¤9¨e¨W¸LÈyÐ#YÑ#YÔ#YÐ Ð Ý˜E¥4Ñ(Ô(ð FµØ•xñ2ô 2ð Fð
  1�Ý#%¤9¨e¨W¸LÈyÐ#YÑ#YÔ#YÐ Ð å#%¤9¨e¨W¸9Ð#EÑ#EÔ#EÐ ØÔ*Ô6ˆKÝœ.Ð)9À{ÐSÑSÔSˆKØ ”N×,Ò,Ñ.Ô.Ð/Ð/r0   c                ó   — dgS ©NrT   © ©Ú_dfs    r.   ú<lambda>z#DaskNamespace.lit.<locals>.<lambda>r   s   € ¨y¨k€ r0   ©Úevaluate_output_namesÚalias_output_namesr7   ©rQ   r   r(   rR   )	r   ÚtypeÚ__name__ÚNotImplementedErrorr   r:   ÚdtypesÚListr3   )r-   rM   rN   Úmsgr`   s   ```  r.   ÚlitzDaskNamespace.litU   sÃ   øøø€ Ý˜UÑ#Ô#ð 	+ØgÕQUÐV[ÑQ\ÔQ\ÔQeÐgÐgˆCÝ% cÑ*Ô*Ð*Ý# E¨4¬=Ô+?Ô+DÑEÔEð 	+ð HˆCÝ% cÑ*Ô*Ð*ð	0ð 	0ð 	0ð 	0ð 	0ð 	0ð 	0ð 	0ð" �zŠzØØ"9Ð"9Ø#Ø”Mð	 ñ 
ô 
ð 	
r0   c                óH   — dd„}|                       |d„ d | j        ¬¦  «        S )	NrQ   r   r(   rR   c                ób   — | j         | j        d                  j                             ¦   «         gS )Nr   )rZ   ÚcolumnsÚsizer\   )rQ   s    r.   r`   zDaskNamespace.len.<locals>.funcx   s(   € àÔ$ R¤Z°¤]Ô3Ô8×BÒBÑDÔDÐEÐEr0   c                ó   — dgS )NÚlenrc   rd   s    r.   rf   z#DaskNamespace.len.<locals>.<lambda>~   s   € ¨u¨g€ r0   rg   rj   )r3   r:   )r-   r`   s     r.   rw   zDaskNamespace.lenw   sH   € ð	Fð 	Fð 	Fð 	Fð �zŠzØØ"5Ð"5Ø#Ø”Mð	 ñ 
ô 
ð 	
r0   ÚexprsÚignore_nullsÚboolc               ól   ‡‡— dˆˆfd„}|                       |t          ‰Ž t          ‰Ž | j        ¬¦  «        S )NrQ   r   r(   rR   c                óŠ   •‡ — t          ‰ gˆ fd„‰D ¦   «         ¢R Ž }‰rd„ |D ¦   «         }t          t          j        |¦  «        gS )Nc              3  ó8   •K  — | ]} |‰¦  «        D ]}|V — ŒŒd S r*   rc   ©Ú.0ÚeÚsrQ   s      €r.   ú	<genexpr>z=DaskNamespace.all_horizontal.<locals>.func.<locals>.<genexpr>…   ó=   øè è € Ð6WÐ6W¸QÐQRÐQRÐSUÑQVÔQVÐ6WÐ6WÈA°qÐ6WÐ6WÐ6WÐ6WÐ6WÐ6WÐ6Wr0   c                óR   — g | ]$}|j         d k    r|n|                     d¦  «        ‘Œ%S )rz   T©rN   Úfillna©r   r�   s     r.   ú
<listcomp>z>DaskNamespace.all_horizontal.<locals>.func.<locals>.<listcomp>Š   s2   € ÐUÐUÐUÈ˜qœw¨&Ò0Ð0˜!˜!°a·h²h¸t±n´nÐUÐUÐUr0   )r   r   ÚoperatorÚand_©rQ   Úseriesrx   ry   s   ` €€r.   r`   z*DaskNamespace.all_horizontal.<locals>.func„   sa   øø€ Ý0°ÐXÐ6WÐ6WÐ6WÐ6WÀ%Ð6WÑ6WÔ6WÐXÐXÐXˆFð ð VàUÐUÈfÐUÑUÔU�Ý�8œ=¨&Ñ1Ô1Ð2Ð2r0   ©Úcallrh   ri   r7   rj   ©r3   r   r   r:   ©r-   ry   rx   r`   s    `` r.   Úall_horizontalzDaskNamespace.all_horizontalƒ   s\   øø€ ð	3ð 	3ð 	3ð 	3ð 	3ð 	3ð 	3ð �zŠzØÝ"?ÀÐ"GÝ9¸5ÐAØ”Mð	 ñ 
ô 
ð 	
r0   c               ól   ‡‡— dˆˆfd„}|                       |t          ‰Ž t          ‰Ž | j        ¬¦  «        S )NrQ   r   r(   rR   c                óŠ   •‡ — t          ‰ gˆ fd„‰D ¦   «         ¢R Ž }‰rd„ |D ¦   «         }t          t          j        |¦  «        gS )Nc              3  ó8   •K  — | ]} |‰¦  «        D ]}|V — ŒŒd S r*   rc   r~   s      €r.   r‚   z=DaskNamespace.any_horizontal.<locals>.func.<locals>.<genexpr>–   rƒ   r0   c                óR   — g | ]$}|j         d k    r|n|                     d¦  «        ‘Œ%S )rz   Fr…   r‡   s     r.   rˆ   z>DaskNamespace.any_horizontal.<locals>.func.<locals>.<listcomp>˜   s2   € ÐVÐVÐVÈ!˜qœw¨&Ò0Ð0˜!˜!°a·h²h¸u±o´oÐVÐVÐVr0   )r   r   r‰   Úor_r‹   s   ` €€r.   r`   z*DaskNamespace.any_horizontal.<locals>.func•   s_   øø€ Ý0°ÐXÐ6WÐ6WÐ6WÐ6WÀ%Ð6WÑ6WÔ6WÐXÐXÐXˆFØð WØVÐVÈvÐVÑVÔV�Ý�8œ<¨Ñ0Ô0Ð1Ð1r0   r�   rj   r�   r�   s    `` r.   Úany_horizontalzDaskNamespace.any_horizontal”   s\   øø€ ð	2ð 	2ð 	2ð 	2ð 	2ð 	2ð 	2ð �zŠzØÝ"?ÀÐ"GÝ9¸5ÐAØ”Mð	 ñ 
ô 
ð 	
r0   c                óh   ‡— dˆfd„}|                       |t          ‰Ž t          ‰Ž | j        ¬¦  «        S )NrQ   r   r(   rR   c                óŽ   •‡ — t          ‰ gˆ fd„‰D ¦   «         ¢R Ž }t          j        |d¬¦  «                             d¬¦  «        gS )Nc              3  ó8   •K  — | ]} |‰¦  «        D ]}|V — ŒŒd S r*   rc   ©r   r3   r�   rQ   s      €r.   r‚   z=DaskNamespace.sum_horizontal.<locals>.func.<locals>.<genexpr>¥   ó8   øè è € Ð>Ð>˜E°E°E¸"±I´IÐ>Ð>¨q�aÐ>Ð>Ð>Ð>Ð>Ð>Ð>r0   é   ©Úaxis)r   rE   ÚconcatÚsum©rQ   rŒ   rx   s   ` €r.   r`   z*DaskNamespace.sum_horizontal.<locals>.func£   sb   øø€ Ý0ØðØ>Ð>Ð>Ð> UÐ>Ñ>Ô>ðð ð ˆFõ ”I˜f¨1Ð-Ñ-Ô-×1Ò1°qÐ1Ñ9Ô9Ð:Ð:r0   r�   rj   r�   ©r-   rx   r`   s    ` r.   Úsum_horizontalzDaskNamespace.sum_horizontal¢   sV   ø€ ð	;ð 	;ð 	;ð 	;ð 	;ð 	;ð �zŠzØÝ"?ÀÐ"GÝ9¸5ÐAØ”Mð	 ñ 
ô 
ð 	
r0   ÚitemsúIterable[DaskLazyFrame]Úhowr    c               óf  — |sd}t          |¦  «        ‚d„ |D ¦   «         }|d         j        }|dk    rÈt          |dd …         d¬¦  «        D ]ƒ\  }}|j        }t          |¦  «        t          |¦  «        k    r||k                         ¦   «         s?d|                     ¦   «         › d|› d	|                     ¦   «         › d
�}t          |¦  «        ‚Œ„t          t          j	        |dd¬¦  «        | j
        ¬¦  «        S |dk    r+t          t          j	        |dd¬¦  «        | j
        ¬¦  «        S t          ‚)NzNo items to concatenatec                ó   — g | ]	}|j         ‘Œ
S rc   )rZ   )r   Úis     r.   rˆ   z(DaskNamespace.concat.<locals>.<listcomp>¶   s   € Ð.Ð.Ð. 1ˆqŒÐ.Ð.Ð.r0   r   Úverticalr�   )Ústartz>unable to vstack, column names don't match:
   - dataframe 0: z
   - dataframe z: ú
Úinner)rŸ   Újoin)r7   ÚdiagonalÚouter)ÚAssertionErrorrt   Ú	enumeraterw   ÚallÚto_listÚ	TypeErrorr   rE   r    r:   rm   )	r-   r¥   r§   rp   ÚdfsÚcols_0rª   rQ   Úcols_currents	            r.   r    zDaskNamespace.concat°   s‚  € ð ð 	&Ø+ˆCÝ  Ñ%Ô%Ð%Ø.Ð.¨Ð.Ñ.Ô.ˆØ�Q””ˆØ�*ÒÐÝ" 3 q r r¤7°!Ð4Ñ4Ô4ð 
)ð 
)‘��2Ø!œz�å˜Ñ&Ô&­#¨f©+¬+Ò5Ð5¸LÈFÒ<R×;WÒ;WÑ;YÔ;YÐ5ðJØ-3¯^ª^Ñ-=Ô-=ðJð Jà*+ðJð Jà/;×/CÒ/CÑ/EÔ/EðJð Jð Jð õ
 $ C™.œ.Ð(ð 6õ !Ý”	˜# A¨GÐ4Ñ4Ô4¸d¼mðñ ô ð ð �*ÒÐÝ Ý”	˜# A¨GÐ4Ñ4Ô4¸d¼mðñ ô ð õ "Ð!r0   c                óh   ‡— dˆfd„}|                       |t          ‰Ž t          ‰Ž | j        ¬¦  «        S )NrQ   r   r(   rR   c                óØ   •‡ — ˆ fd„‰D ¦   «         }t          ‰ g|¢R Ž }d„ |D ¦   «         }d„ |D ¦   «         }t          d„ |¦  «        }t          d„ |¦  «        }t          d||z  ¦  «        gS )Nc                ó0   •— g | ]} |‰¦  «        D ]}|‘ŒŒS rc   rc   r›   s      €r.   rˆ   z?DaskNamespace.mean_horizontal.<locals>.func.<locals>.<listcomp>Ð   s.   ø€ ÐDÐDÐD %¸%¸%À¹)¼)ÐDÐD°Q˜AÐDÐDÐDÐDr0   c              3  ó@   K  — | ]}|                      d ¦  «        V — ŒdS )r   N©r†   r‡   s     r.   r‚   z>DaskNamespace.mean_horizontal.<locals>.func.<locals>.<genexpr>Ò   s,   è è € Ð3Ð3 a�a—h’h˜q‘k”kÐ3Ð3Ð3Ð3Ð3Ð3r0   c              3  óD   K  — | ]}d |                      ¦   «         z
  V — ŒdS )r�   N©Úisnar‡   s     r.   r‚   z>DaskNamespace.mean_horizontal.<locals>.func.<locals>.<genexpr>Ó   s.   è è € Ð4Ð4 q�a˜!Ÿ&š&™(œ(‘lÐ4Ð4Ð4Ð4Ð4Ð4r0   c                ó   — | |z   S r*   rc   ©ÚxÚys     r.   rf   z=DaskNamespace.mean_horizontal.<locals>.func.<locals>.<lambda>Ô   ó
   €  a¨!¡e€ r0   c                ó   — | |z   S r*   rc   rÃ   s     r.   rf   z=DaskNamespace.mean_horizontal.<locals>.func.<locals>.<lambda>Õ   rÆ   r0   z	dx.Series)r   r   r
   )rQ   Úexpr_resultsÚalignedrŒ   Únon_naÚnumÚdenrx   s   `      €r.   r`   z+DaskNamespace.mean_horizontal.<locals>.funcÏ   s•   øø€ ØDÐDÐDÐD¨5ÐDÑDÔDˆLÝ1°"ÐD°|ÐDÐDÐDˆGØ3Ð3¨7Ð3Ñ3Ô3ˆFØ4Ð4¨GÐ4Ñ4Ô4ˆFÝÐ+Ð+¨VÑ4Ô4ˆCÝÐ+Ð+¨VÑ4Ô4ˆCÝ˜ c¨C¡iÑ0Ô0Ð1Ð1r0   r�   rj   r�   r£   s    ` r.   Úmean_horizontalzDaskNamespace.mean_horizontalÎ   sV   ø€ ð	2ð 	2ð 	2ð 	2ð 	2ð 	2ð �zŠzØÝ"?ÀÐ"GÝ9¸5ÐAØ”Mð	 ñ 
ô 
ð 	
r0   c                óh   ‡— dˆfd„}|                       |t          ‰Ž t          ‰Ž | j        ¬¦  «        S )NrQ   r   r(   rR   c                óŽ   •‡ — t          ‰ gˆ fd„‰D ¦   «         ¢R Ž }t          j        |d¬¦  «                             d¬¦  «        gS )Nc              3  ó8   •K  — | ]} |‰¦  «        D ]}|V — ŒŒd S r*   rc   r›   s      €r.   r‚   z=DaskNamespace.min_horizontal.<locals>.func.<locals>.<genexpr>â   rœ   r0   r�   rž   )r   rE   r    Úminr¢   s   ` €r.   r`   z*DaskNamespace.min_horizontal.<locals>.funcà   ób   øø€ Ý0ØðØ>Ð>Ð>Ð> UÐ>Ñ>Ô>ðð ð ˆFõ ”I˜f¨1Ð-Ñ-Ô-×1Ò1°qÐ1Ñ9Ô9Ð:Ð:r0   r�   rj   r�   r£   s    ` r.   Úmin_horizontalzDaskNamespace.min_horizontalß   óV   ø€ ð	;ð 	;ð 	;ð 	;ð 	;ð 	;ð �zŠzØÝ"?ÀÐ"GÝ9¸5ÐAØ”Mð	 ñ 
ô 
ð 	
r0   c                óh   ‡— dˆfd„}|                       |t          ‰Ž t          ‰Ž | j        ¬¦  «        S )NrQ   r   r(   rR   c                óŽ   •‡ — t          ‰ gˆ fd„‰D ¦   «         ¢R Ž }t          j        |d¬¦  «                             d¬¦  «        gS )Nc              3  ó8   •K  — | ]} |‰¦  «        D ]}|V — ŒŒd S r*   rc   r›   s      €r.   r‚   z=DaskNamespace.max_horizontal.<locals>.func.<locals>.<genexpr>ñ   rœ   r0   r�   rž   )r   rE   r    Úmaxr¢   s   ` €r.   r`   z*DaskNamespace.max_horizontal.<locals>.funcï   rÒ   r0   r�   rj   r�   r£   s    ` r.   Úmax_horizontalzDaskNamespace.max_horizontalî   rÔ   r0   c          	     óð   ‡‡‡‡— t          t          | j        | j        ¬¦  «        Šdˆˆˆˆfd„}|                      |t          ‰d         dd	„ ¦  «        t          ‰d         d
d ¦  «        | j        ¬¦  «        S )N)r7   ÚimplementationrQ   r   r(   rR   c                óò  •‡ — t          ‰ gˆ fd„‰D ¦   «         ¢R Ž }ˆfd„|D ¦   «         }d„ |D ¦   «         }‰sCt          t          j        |¦  «        }t          ˆfd„|¦  «                             | d ¦  «        }n{d„ t          ||d¬¦  «        D ¦   «         ^}}|d          }|}t          |d	d …         |d¬¦  «        D ]7\  }	}
|	 }||z                       ‰d
dœt          ¬¦  «        }||z   |
z   }||z  }Œ8|gS )Nc              3  ó8   •K  — | ]} |‰¦  «        D ]}|V — ŒŒd S r*   rc   r›   s      €r.   r‚   z9DaskNamespace.concat_str.<locals>.func.<locals>.<genexpr>  rœ   r0   c              3  óò   •K  — | ]q} ‰|j         ¦  «                             ¦   «         r1|                     t          ¦  «        j                             ¦   «         n|                     t          ¦  «        V — Œrd S r*   )rN   Ú
is_booleanÚastyper?   Úlower)r   r�   Úto_nw_dtypes     €r.   r‚   z9DaskNamespace.concat_str.<locals>.func.<locals>.<genexpr>  s�   øè è € ð ð ð ð �;˜qœwÑ'Ô'×2Ò2Ñ4Ô4ð#�—’�‘”Ô!×'Ò'Ñ)Ô)Ð)à—X’X�c‘]”]ðð ð ð ð ð r0   c                ó6   — g | ]}|                      ¦   «         ‘ŒS rc   rÀ   r‡   s     r.   rˆ   z:DaskNamespace.concat_str.<locals>.func.<locals>.<listcomp>  s    € Ð8Ð8Ð8 a˜Ÿš™œÐ8Ð8Ð8r0   c                ó   •— | ‰z   |z   S r*   rc   )rÄ   rÅ   r=   s     €r.   rf   z8DaskNamespace.concat_str.<locals>.func.<locals>.<lambda>  s   ø€ ¨Q°©]¸QÑ->€ r0   c                óB   — g | ]\  }}|                      | d ¦  «        ‘ŒS )Ú )Úwhere)r   r�   Únms      r.   rˆ   z:DaskNamespace.concat_str.<locals>.func.<locals>.<listcomp>  s9   € ð 'ð 'ð 'Ù).¨¨B�A—G’G˜R˜C Ñ$Ô$ð'ð 'ð 'r0   T)Ústrictr   r�   ræ   )TF)Úmeta)r   r   r‰   r–   rç   ÚzipÚmapr?   )rQ   rÈ   rŒ   Ú	null_maskÚnull_mask_resultÚresultÚ
init_valueÚvaluesÚseenrè   rM   Únot_nullÚ
separatorsrx   ry   r=   râ   s   `            €€€€r.   r`   z&DaskNamespace.concat_str.<locals>.func  s‡  øø€ Ý6ØðØ>Ð>Ð>Ð> UÐ>Ñ>Ô>ðð ð ˆLðð ð ð ð &ð	ñ ô ˆFð 9Ð8¨<Ð8Ñ8Ô8ˆIàð +Ý#)­(¬,¸	Ñ#BÔ#BÐ ÝÐ >Ð >Ð >Ð >ÀÑGÔG×MÒMØ%Ð% tñô ��ð'ð 'Ý25°f¸iÐPTÐ2UÑ2UÔ2Uð'ñ 'ô 'Ð#�
˜Vð " !œ�}�Ø#�Ý!$ Y¨q¨r¨r¤]°FÀ4Ð!HÑ!HÔ!Hð +ð +‘I�B˜Ø "˜s�HØ"&¨¡/×!6Ò!6Ø(°Ð4Ð4½3ð "7ñ "ô "�Jð $ jÑ0°5Ñ8�FØ (™?�D�Dà�8ˆOr0   r   Ú_evaluate_output_namesc                ó   — dgS rb   rc   rd   s    r.   rf   z*DaskNamespace.concat_str.<locals>.<lambda>0  s   € ÀÀ€ r0   Ú_alias_output_namesr�   rj   )r   r   r:   r'   r3   Úgetattr)r-   r=   ry   rx   r`   râ   s    ``` @r.   Ú
concat_strzDaskNamespace.concat_strý   s°   øøøø€ õ Ý$Ø”MØÔ/ð
ñ 
ô 
ˆð%	ð %	ð %	ð %	ð %	ð %	ð %	ð %	ð %	ðN �zŠzØÝ")Ø�a”Ð2Ð4KÐ4Kñ#ô #õ  ' u¨Q¤xÐ1FÈÑMÔMØ”Mð ñ 
ô 
ð 	
r0   c                óh   ‡— dˆfd„}|                       |t          ‰Ž t          ‰Ž | j        ¬¦  «        S )NrQ   r   r(   rR   c                ó\   •‡ — t          ‰ gˆ fd„‰D ¦   «         ¢R Ž }t          d„ |¦  «        gS )Nc              3  ó8   •K  — | ]} |‰¦  «        D ]}|V — ŒŒd S r*   rc   r›   s      €r.   r‚   z7DaskNamespace.coalesce.<locals>.func.<locals>.<genexpr>9  rœ   r0   c                ó,   — |                       |¦  «        S r*   r¾   rÃ   s     r.   rf   z6DaskNamespace.coalesce.<locals>.func.<locals>.<lambda>;  s   € ¨¯ª°©¬€ r0   )r   r   r¢   s   ` €r.   r`   z$DaskNamespace.coalesce.<locals>.func7  sP   øø€ Ý0ØðØ>Ð>Ð>Ð> UÐ>Ñ>Ô>ðð ð ˆFõ Ð3Ð3°VÑ<Ô<Ð=Ð=r0   r�   rj   r�   r£   s    ` r.   ÚcoalescezDaskNamespace.coalesce6  sV   ø€ ð	>ð 	>ð 	>ð 	>ð 	>ð 	>ð �zŠzØÝ"?ÀÐ"GÝ9¸5ÐAØ”Mð	 ñ 
ô 
ð 	
r0   NÚ	predicateÚthenÚ	otherwiseúDaskExpr | Nonec           	     ó’   ‡‡‡— d
ˆˆˆfd„}|                       |t          ‰dd„ ¦  «        t          ‰dd ¦  «        | j        ¬	¦  «        S )NrQ   r   r(   rR   c                ó€  •— |                       ‰	¦  «        }‰�|                       ‰¦  «        n‰}|                       ‰¦  «        }t          d„ ‰€‰‰	fn‰‰	‰fD ¦   «         ¦  «        rP|                      |                     ¦   «         ¦  «        }|                       ‰                     ¦   «         ¦  «        }|} ‰€:t          | ||¦  «        \  }}t          ||¦  «         |                     |¦  «        gS t          | |||¦  «        \  }}}t          ||¦  «         t          ||¦  «         |                     ||¦  «        gS )Nc              3  ó.   K  — | ]}|j         j        V — Œd S r*   )Ú	_metadataÚis_scalar_like)r   rÄ   s     r.   r‚   z8DaskNamespace.when_then.<locals>.func.<locals>.<genexpr>R  s;   è è € ð ð àð ”Ô*ðð ð ð ð ð r0   )Ú_evaluate_single_output_exprr´   Ú_with_nativeÚto_frameÚ	broadcastr   r   rç   )
rQ   Ú
then_valueÚotherwise_valueÚ	conditionÚnew_dfÚthen_seriesÚotherwise_seriesr  rÿ   r   s
          €€€r.   r`   z%DaskNamespace.when_then.<locals>.funcG  s‡  ø€ Ø×8Ò8¸Ñ>Ô>ˆJð Ð(ð ×/Ò/°	Ñ:Ô:Ð:àð ð ×7Ò7¸	ÑBÔBˆIõ ð ð ð !Ð(ð  Ð%Ð%à# T¨9Ð5ðñ ô ñ ô ð 
ð Ÿš¨×);Ò);Ñ)=Ô)=Ñ>Ô>�Ø×;Ò;¸I×<OÒ<OÑ<QÔ<QÑRÔR�	Ø�àÐ Ý)DØ˜	 :ñ*ô *Ñ&�	˜;õ # 9¨kÑ:Ô:Ð:Ø#×)Ò)¨)Ñ4Ô4Ð5Ð5Ý7RØ�I˜z¨?ñ8ô 8Ñ4ˆI�{Ð$4õ ˜y¨+Ñ6Ô6Ð6Ý˜yÐ*:Ñ;Ô;Ð;Ø×%Ò% iÐ1AÑBÔBÐCÐCr0   rõ   c                ó   — dgS rb   rc   rd   s    r.   rf   z)DaskNamespace.when_then.<locals>.<lambda>n  s   € ¸Y¸K€ r0   r÷   r�   rj   )r3   rø   r:   )r-   rÿ   r   r  r`   s    ``` r.   Ú	when_thenzDaskNamespace.when_thenD  s†   øøø€ ð"	Dð "	Dð "	Dð "	Dð "	Dð "	Dð "	Dð "	DðH �zŠzØÝ")ØÐ.Ð0GÐ0Gñ#ô #õ  ' tÐ-BÀDÑIÔIØ”Mð ñ 
ô 
ð 	
r0   ÚaÚbÚmethodr!   c               óŒ   ‡‡‡— dˆˆˆfd„}|                       |t          ‰‰¦  «        t          ‰‰¦  «        | j        ¬¦  «        S )NrQ   r   r(   rR   c                ó¬   •— |                       ‰¦  «        }|                       ‰¦  «        }|                     |‰¬¦  «                             ¦   «         gS )N)r  )r  Úcorrr\   )rQ   Úa_Úb_r  r  r  s      €€€r.   r`   z DaskNamespace.corr.<locals>.funcu  sN   ø€ Ø×0Ò0°Ñ3Ô3ˆBØ×0Ò0°Ñ3Ô3ˆBØ—G’G˜B v�GÑ.Ô.×8Ò8Ñ:Ô:Ð;Ð;r0   r�   rj   r�   )r-   r  r  r  r`   s    ``` r.   r  zDaskNamespace.corrt  sl   øøø€ ð	<ð 	<ð 	<ð 	<ð 	<ð 	<ð 	<ð 	<ð
 �zŠzØÝ"?ÀÀ1Ñ"EÔ"EÝ9¸!¸QÑ?Ô?Ø”Mð	 ñ 
ô 
ð 	
r0   ÚddofÚintc               óŒ   ‡‡‡— dˆˆˆfd„}|                       |t          ‰‰¦  «        t          ‰‰¦  «        | j        ¬¦  «        S )NrQ   r   r(   rR   c                ó  •— |                       ‰¦  «        }|                       ‰¦  «        }|                     |¦  «                             ¦   «         }‰	dk    r|gS |                     ¦   «          |                     ¦   «          z                       ¦   «                              ¦   «         }|‰	z
  }||dz
  z  |                     |dk    ¦  «        z  }‰	dk    r|                     |dk    d¦  «        }|                     |j        ¦  «        gS )Nr�   r   g        )r  Úcovr\   rÁ   r¡   rç   ÚrenamerU   )
rQ   r  r  Úcov_sampÚ	n_samplesÚdenominatorrï   r  r  r  s
          €€€r.   r`   zDaskNamespace.cov.<locals>.func‚  sô   ø€ Ø×0Ò0°Ñ3Ô3ˆBØ×0Ò0°Ñ3Ô3ˆBð —v’v˜b‘z”z×+Ò+Ñ-Ô-ˆHØ�qŠyˆyØ �zÐ!àŸ'š'™)œ)˜ r§w¢w¡y¤y jÑ0×5Ò5Ñ7Ô7×AÒAÑCÔCˆIØ# dÑ*ˆKØ ¨Q¡Ñ/°+×2CÒ2CÀKÐRSÂOÑ2TÔ2TÑTˆFØ�qŠyˆyØŸš i°1¢n°cÑ:Ô:�Ø—M’M "¤'Ñ*Ô*Ð+Ð+r0   r�   rj   r�   )r-   r  r  r  r`   s    ``` r.   r   zDaskNamespace.cov�  sl   øøø€ ð	,ð 	,ð 	,ð 	,ð 	,ð 	,ð 	,ð 	,ð  �zŠzØÝ"?ÀÀ1Ñ"EÔ"EÝ9¸!¸QÑ?Ô?Ø”Mð	 ñ 
ô 
ð 	
r0   )r(   r   )r(   r1   )r(   r4   )r7   r   r(   r8   )r>   r$   r=   r?   r@   r	   r(   r   )r>   r$   r@   r	   r(   r   )rM   r#   rN   rO   r(   r   )r(   r   )rx   r   ry   rz   r(   r   )rx   r   r(   r   )r¥   r¦   r§   r    r(   r   )rx   r   r=   r?   ry   rz   r(   r   r*   )rÿ   r   r   r   r  r  r(   r   )r  r   r  r   r  r!   r(   r   )r  r   r  r   r  r  r(   r   )rl   Ú
__module__Ú__qualname__r   ÚDASKr'   Ú__annotations__Úpropertyr/   r3   r6   r;   rI   rL   rq   rw   r‘   r—   r¤   r    rÍ   rÓ   rÙ   rù   rþ   r  r  r   r   ÚstructÚlistrc   r0   r.   r&   r&   1   s  € € € € € € ð '5Ô&9€OÐ9Ð9Ð9Ñ9àð:ð :ð :ñ „Xð:ð ðð ð ñ „Xðð ðð ð ñ „Xðð ð  ð  ð  ð =@ðAð Að Að Að Að Að
ð 
ð 
ð 
ð 
ð  
ð  
ð  
ðD

ð 

ð 

ð 

ð
ð 
ð 
ð 
ð"
ð 
ð 
ð 
ð
ð 
ð 
ð 
ð"ð "ð "ð "ð<
ð 
ð 
ð 
ð"
ð 
ð 
ð 
ð
ð 
ð 
ð 
ð7
ð 7
ð 7
ð 7
ðr
ð 
ð 
ð 
ð QUð.
ð .
ð .
ð .
ð .
ð`
ð 
ð 
ð 
ð
ð 
ð 
ð 
ð0 ˆ_ÑÔ€FØˆ?ÑÔ€D€D€Dr0   r&   )8Ú
__future__r   r‰   r   r   Ú	functoolsr   r   Útypingr   r	   r
   Údask.dataframeÚ	dataframerE   ÚpandasrW   Únarwhals._compliantr   r   Únarwhals._dask.dataframer   Únarwhals._dask.exprr   Únarwhals._dask.selectorsr   Únarwhals._dask.utilsr   r   r   Únarwhals._expression_parsingr   r   Únarwhals._pandas_like.utilsr   Únarwhals._utilsr   r   r   r   r   r   Úcollections.abcr   Údask.dataframe.dask_exprÚ	dask_exprÚdxr   Únarwhals.typingr    r!   r"   r#   r$   Ú	DataFramer&   rc   r0   r.   ú<module>r@     s‚  ðØ "Ð "Ð "Ð "Ð "Ð "à €€€Ø #Ð #Ð #Ð #Ð #Ð #Ð #Ð #Ø %Ð %Ð %Ð %Ð %Ð %Ð %Ð %Ø +Ð +Ð +Ð +Ð +Ð +Ð +Ð +Ð +Ð +à Ð Ð Ð Ð Ð Ø Ð Ð Ð à EÐ EÐ EÐ EÐ EÐ EÐ EÐ EØ 2Ð 2Ð 2Ð 2Ð 2Ð 2Ø (Ð (Ð (Ð (Ð (Ð (Ø :Ð :Ð :Ð :Ð :Ð :ðð ð ð ð ð ð ð ð ð ð
ð ð ð ð ð ð ð ð AÐ @Ð @Ð @Ð @Ð @ðð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð ð Ø(Ð(Ð(Ð(Ð(Ð(à)Ð)Ð)Ð)Ð)Ð)Ð)Ð)Ð)à'Ð'Ð'Ð'Ð'Ð'ðð ð ð ð ð ð ð ð ð ð ð ð ð ðið ið ið ið iØ�- ¨2¬<Ð7Ô8Ø˜=¨(Ð2Ô3ñiô ið ið ið ir0   