Ë
    <Dj  ã                   ó4   — d Z ddlZddlZddlmZ deddfd„Zy)z*Tests related to the `DataIter` interface.é    N)ÚtestingÚdeviceÚreturnc                 óà  — t        j                  ddd«      \  }}}| j                  d«      rt        j                  dddd«      \  }}nt        j                  dddd«      \  }}t        j                  ddd	d«      \  }}|||g}|||g}	| j                  d«      r"d
dl}
|D cg c]  }|
j                  |«      ‘Œ }}t        j                  ||	dd«      }t        j                  |«      }t        j                  dddd«      \  }}|||g}|||g}	t        j                  |d
¬«      }t        j                  |	d
¬«      }t        j                  ||«      }t        j                  ||«      sJ ‚yc c}w )zCheck QDM with mixed batches.é€   é   FÚcudaé   gš™™™™™¹?Ti   gÍÌÌÌÌÌì?r   N)Úaxis)ÚtmÚmake_regressionÚ
startswithÚmake_sparse_regressionÚcupyÚarrayÚIteratorForTestÚxgboostÚQuantileDMatrixÚnpÚconcatenateÚpredictor_equal)r   ÚX_0Úy_0Ú_ÚX_1Úy_1ÚX_2Úy_2ÚXÚyÚcpÚbatchÚitÚXy_0ÚX_arrÚy_arrÚXy_1s                    ú]C:\Crop_Prediction\Backend\crop-ai-system\venv\Lib\site-packages\xgboost/testing/data_iter.pyÚrun_mixed_sparsityr)   	   s]  € ä×$Ñ$ S¨"¨eÓ4K€CˆˆaØ×Ñ˜Ô Ü×,Ñ,¨S°"°c¸4Ó@‰ˆ‰Sä×,Ñ,¨S°"°c¸5ÓA‰ˆˆSÜ×(Ñ(¨¨b°#°tÓ<H€CˆØ	ˆc3ˆ€AØ	ˆc3ˆ€Aà×Ñ˜Ô Ûà*+Ö, ˆRX‰Xe_Ð,ˆÐ,ä	×	Ñ	˜A˜q $¨Ó	-€BÜ×"Ñ" 2Ó&€Dä×(Ñ(¨¨b°#°tÓ<H€CˆØ	ˆc3ˆ€AØ	ˆc3ˆ€AÜN‰N˜1 1Ô%€EÜN‰N˜1 1Ô%€EÜ×"Ñ" 5¨%Ó0€Dä×Ñ˜d DÔ)Ñ)Ð)ùò -s   Â"E+)Ú__doc__Únumpyr   r   r   r   Ústrr)   © ó    r(   ú<module>r/      s$   ðÙ 0ã ã Ý !ð*˜sð * tô *r.   