i ddlmZddlmZddlmZddlmZddl m Z m Z ddded e d e d ed ejeeff d Zy))Any) DataFrame)numpy) IndexOrder)ArrayListfortran)orderdfwritabletargetr returnc|jjD]#\}}|tk(sd|d|d}t|d|jdk(rG|jj dt k(r#||jdj|}n|j||}|jtk(rd |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 U/var/www/pod-logistic/pod-ai/venv/lib/python3.12/site-packages/polars/ml/utilities.pyframe_to_numpyr s))//#+B :/vT&A[\CC.d *+  xx1}))+A.%7A((((;kk85k9 yyF,VH4degenendqrn JN)typingrpolarsrpolars._dependenciesrnppolars._typingrpolars.datatypesrrboolstrndarrayrr rr+s[,%("      ZZS r