^i UdZddlmZmZmZddlZddlmZddl m Z ddl m Z edZ eed <d dd d e d ededededdf dZde de defdZy)zHelpers for test code.)AnyLiteral TypeAliasN) import_cupy)DMatrix)_is_cupy_alike)cpucudaDevicegHz>)rtolatoldeviceabr rreturnc|dk(r:t|s/t|s$tjj||||yt }|jj||||y)z)Dispatch the assert_allclose for devices.r )rr N)r nptestingassert_allcloser)rrrr rcps V/var/www/html/hubwallet-dev/venv/lib/python3.12/site-packages/xgboost/testing/utils.pyrrsV~a09J ""1ad"> ] ""1ad">lhsrhscN|j}|j}ttj|j|jtj|j |j tj|j |j fS)z9Assert whether two DMatrices contain the same predictors.)get_dataallr array_equaldataindicesindptr)rrlcsrrcsrs rpredictor_equalr%sl <<>D <<>D  NN499dii 0 NN4<< 6 NN4;; 4  r)__doc__typingrrrnumpyrcompatrcorerr r r __annotations__floatrboolr%rrr/s** !M* *6:? ??"?-2?AF? ?  w 4 r