o i @srddlmZddlmZddlmZddlmZddl m Z m Z ddded e d e d ed ejeeff d dZdS))Any) DataFrame)numpy) IndexOrder)ArrayListfortran)orderdfwritabletargetr returncCs|jD]\}}|tkrd|d|d}t|dq|jdkr6|jdtkr6||jdj|d}n|j||d}|j t krOd |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) ) Zschemaitemsr TypeErrorwidthZdtypesrcolumnsZto_numpyZdtypeobject)r r r r nmtpmsgZarrrJ/home/app/Keep/.python/lib/python3.10/site-packages/polars/ml/utilities.pyframe_to_numpy s  rN)typingrZpolarsrZpolars._dependenciesrnpZpolars._typingrZpolars.datatypesrrboolstrZndarrayrrrrrs"