o g @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)) ValueErrorrTrrDTYPErnpma isMaskedArraymaskarraydatar rasarrayrndimshape) csgraphdirectedr csr_output dense_output copy_if_densecopy_if_sparse null_value_innull_value_outrr rr$^/home/ubuntu/cloudmapper/venv/lib/python3.10/site-packages/scipy/sparse/csgraph/_validation.pyvalidate_graph sB      r&)numpyr scipy.sparserrr_toolsrrrr float64rinfr&r$r$r$r%s