
    bi                    4    d dl mZ d Zd Zd Zd Zd Zd Zy)	    )annotationsc                      y N )argss    I/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/dask/bag/chunk.pybarrierr	      s        c                    | |   S )znLike :func:`operator.getitem`, but allows setting key using partial
    ``partial(chunk.getitem, key=key)
    r   )xkeys     r   getitemr      s     S6Mr
   c                     | ||d         S )N   r   )combineaccr   s      r   foldby_combine2r      s    3!r
   c                "     | ||             | fS r   r   )r   hashgroupers      r   groupby_tasks_group_hashr      s    
Qr
   c                J    d\  }}}| D ]  }||dz  z  }||z  }|dz  } |||fS )N)        r   r      r   r   )seqsquarestotalnr   s        r   	var_chunkr      sK    #GUA 1a4
	Q E1r
   c                    t        t        |        \  }}}t        t        |            t        t        |            t        |      }} }||z  | |z  dz  z
  }||z  ||z
  z  S )Nr   )listzipfloatsum)r   ddofr   totalscountsx2r   results           r   var_aggregater*       sb    "37mGVVS\"E#f+$6F1B1fQ1$FA:T""r
   N)
__future__r   r	   r   r   r   r   r*   r   r
   r   <module>r,      s&    "#r
   