U
    Åmœd«  ã                   @   s>   d dl Zzd dlmZ W n ek
r0   dZY nX dd„ ZdS )é    Nc                 C   sV   t | ƒttfkrLtdk	r:tdd„ | D ƒƒr:tj| ddiŽS tdd„ | D ƒƒS t | ¡S )z'Convert distributed arrays to ndarrays.Nc                 s   s   | ]}t |tjƒV  qd S ©N)Ú
isinstanceÚdaZArray©Ú.0Zarr© r   úZ/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/scanpy/preprocessing/_distributed.pyÚ	<genexpr>   s     z)materialize_as_ndarray.<locals>.<genexpr>ÚsyncTc                 s   s   | ]}t  |¡V  qd S r   )ÚnpÚasarrayr   r   r   r   r	      s     )ÚtypeÚlistÚtupler   ÚanyZcomputer   r   )Úar   r   r   Úmaterialize_as_ndarray
   s
    r   )Únumpyr   Z
dask.arrayÚarrayr   ÚImportErrorr   r   r   r   r   Ú<module>   s
   
