gdZddlmZmZmZddlZddlmZdee ee fdee ejejffdZ y)z| Module to load the minimal information from a Stan CSV file. Only the header row and data are read, no metadata is parsed. )ListTupleUnionN filenamesreturnct|ts|g}d}tt|Dcgc]}d}}t |D]\}}t |d5}|j x}jdr# |j x}jdr#|dk(r|}n||k7r tdtj|dd||<ddd|jtj|d fScc}w#1swYxYw) ad Reads CSV files like those produced by Stan, returning the header and data. If multiple files are given, the data is stacked along the first axis, so in typical usage, the shape of the returned data will be ``(n_chains, n_samples, n_params)``. Parameters ---------- filenames : Union[str, List[str]] Path to the CSV file(s) to read. Returns ------- Tuple[str, npt.NDArray[np.float64]] The header row and data from the CSV file(s). Raises ------ ValueError If multiple files are given and the headers do not match between them. Nr#zHeaders do not match,) delimitercommentsr)axis) isinstancelistrangelen enumerateopenreadline startswith ValueErrornploadtxtstripstack)rheader_dataiffd file_headers G/var/www/dash_apps/app1/venv/lib/python3.12/site-packages/stanio/csv.pyread_csvr% s 0 i &K F9>s9~9N*OA4*OD*O)$B1 !S\ BR"$++-/;;;C@#%++-/;;;C@|$;& !788jjsSADG B BB <<>288Dq1 11+P B Bs C=AD3DD ) __doc__typingrrrnumpyr numpy.typingnptstrNDArrayfloat64r%r$r0sO&%'2c49n-'2%S[[=T8T2U'2r/