xf zddlZddlmZmZddlmZmZmZm Z ejZ e dddddejddf dZ y)N) csr_matrixissparse)csgraph_to_densecsgraph_from_densecsgraph_masked_from_densecsgraph_from_maskedTFc |s |s td|s&t|r|jdk(r |j}t|r#|rt |t |}nt ||}ntjj|rF|r8|j} tj|jt |}||| <ndt|}nX|rGt|||| | }|j} tj|jt }||| <nt!||| | }|j"dk7r td |j$d |j$d k7r td |S) z7Routine for validation and conversion of csgraph inputsz*Internal: dense or csr output must be truecsc)dtypecopy) null_value)r rnan_null infinity_null)r )rrrz#compressed-sparse graph must be 2-Drrz,compressed-sparse graph must be shape (N, N)) ValueErrorrformatTrDTYPErnpma isMaskedArraymaskarraydatar rasarrayrndimshape) csgraphdirectedr csr_output dense_output copy_if_densecopy_if_sparse null_value_innull_value_outrrrs B/usr/lib/python3/dist-packages/scipy/sparse/csgraph/_validation.pyvalidate_graphr( s> ,EFF (7+%0G))  NKG&w>JG   W % <K MG <??}}Q7==++GHH N)numpyr scipy.sparserr_toolsrrrr float64rinfr(r)r'r0sB-33 -2"!&u!"266!% /r)