sii UdZddlmZmZmZddlZddlmZddl m Z ddl m Z edZ eed <d dd d e d ededededdf dZde de defdZdS)zHelpers for test code.)AnyLiteral TypeAliasN) import_cupy)DMatrix)_is_cupy_alike)cpucudaDevicegHz>)rtolatoldeviceabr rreturnc|dkrCt|s4t|s%tj||||dSt }|j||||dS)z)Dispatch the assert_allclose for devices.r )rr N)r nptestingassert_allcloser)rrrr rcps [/var/www/html/bet.cuttalo.com/ml/venv/lib/python3.11/site-packages/xgboost/testing/utils.pyrrsx~a009J9J ""1ad">>>>> ]] ""1ad">>>>>lhsrhsc$|}|}ttj|j|jtj|j|jtj|j|jfS)z9Assert whether two DMatrices contain the same predictors.)get_dataallr array_equaldataindicesindptr)rrlcsrrcsrs rpredictor_equalr%sl <<>>D <<>>D  N49di 0 0 N4< 6 6 N4; 4 4   r)__doc__typingrrrnumpyrcompatrcorerr r r __annotations__floatrboolr%rrr/s ********** !!!!!!M* ***6:??? ??"?-2?AF? ????  w 4      r