o
    id                     @   s^   d dl Zd dlmZ d dlmZ zd dlZ	dZ
W n ey#   dZ
Y nw eje Zdd ZdS )    N)ndarray)spmatrixTFc                  G   s$   | D ]}t |ttfrt  S qtS )a  Returns the array module for arguments.

    This function is used to implement CPU/GPU generic code. If at least one of
    the arguments is a :class:`cupy.ndarray` object, the :mod:`cupyx.scipy`
    module is returned.

    Args:
        args: Values to determine whether NumPy or CuPy should be used.

    Returns:
        module: :mod:`cupyx.scipy` or :mod:`scipy` is returned based on the
        types of the arguments.

    )
isinstance_ndarray	_spmatrix_cupyx_scipy_scipy)argsarg r   Q/home/ubuntu/veenaModal/venv/lib/python3.10/site-packages/cupyx/scipy/__init__.pyget_array_module   s
   r   )sys_sys
cupy._corer   r   cupyx.scipy.sparse._baser   r   scipyr   _scipy_availableImportErrormodules__name__r   r   r   r   r   r   <module>   s    
