o Eb @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|st|r|j}t|r&|rt|t|d}nHt||d}nAtj |rD|r?|j } tj |j t|d}||| <n(t |}n#|r_t|||| | d}|j } tj|j td}||| <nt||| | d}|jdkrptd|jd |jd kr~td |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)) ValueErrorrTrrDTYPErnpZmaZ isMaskedArraymaskZarraydatar rZasarrayrndimshape) ZcsgraphZdirectedr Z csr_outputZ dense_outputZ copy_if_denseZcopy_if_sparseZ null_value_inZnull_value_outrr rrB/usr/lib/python3/dist-packages/scipy/sparse/csgraph/_validation.pyvalidate_graph sB      r)ZnumpyrZ scipy.sparserrrZ_toolsrrrr Zfloat64rinfrrrrrs