U /d @s\ddlZddlmZmZmZddlmZmZm Z m Z ej Z e dddddej ddf ddZdS) N) csr_matrix isspmatrixisspmatrix_csc)csgraph_to_densecsgraph_from_densecsgraph_masked_from_densecsgraph_from_maskedTFc Cs|s|std|s"t|r"|j}t|rL|r>t|t|d}qt||d}ntj |r|r~|j } tj |j t|d}||| <qt |}nF|rt|||| | d}|j } tj|j td}||| <nt||| | d}|jdkrtd|jd |jd krtd |S) z7Routine for validation and conversion of csgraph inputsz*Internal: dense or csr output must be true)dtypecopy) null_value)r r nan_null infinity_null)r )r rr z#compressed-sparse graph must be 2-Drrz,compressed-sparse graph must be shape (N, N)) ValueErrorrTrrDTYPErnpmaZ isMaskedArraymaskarraydatar rZasarrayrndimshape) ZcsgraphZdirectedr Z csr_outputZ dense_outputZ copy_if_denseZcopy_if_sparseZ null_value_inZnull_value_outrr rrD/tmp/pip-unpacked-wheel-9gxwnfpp/scipy/sparse/csgraph/_validation.pyvalidate_graph sB      r)ZnumpyrZ scipy.sparserrrZ_toolsrrrr Zfloat64rinfrrrrrs