q-Ph ddlmZddlmZddlmZddlmZmZddl m Z ddded e d e d ed e jeeff d ZdS))Any) DataFrame) IndexOrder)ArrayList)numpyfortran)orderdfwritabletargetr returnc|jD])\}}|tkrd|d|d}t|d*|jdkrP|jdt kr(||jd|}n|||}|j tkrd |d |j}t||S) zAConvert a DataFrame to a NumPy array for use with Jax or PyTorch.zcannot convert List column z to z (use Array dtype instead)Nr)r )r r zcannot convert DataFrame to z/ (mixed type columns result in `object` dtype) ) schemaitemsr TypeErrorwidthdtypesrcolumnsto_numpydtypeobject)r r r r nmtpmsgarrs S/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/polars/ml/utilities.pyframe_to_numpyr s)//##++B ::\\\&\\\CC..d *  x1}}))++A.%77A((((;;kk85k99 yFrVrregenrrnn JN)typingrpolarsrpolars._typingrpolars.datatypesrrpolars.dependenciesrnpboolstrndarrayrr rr+s%%%%%%((((((((++++++"      ZS r