g dZ ddlZeeeej jdZedkre ddZ ddl m Z m Z ddlZd e ee fd e ee ffd Zd e d e fd Zde ee fd efdZdede ee fd dfdZy#dZ ddlZYMxYw)z' Utilities for writing Stan Json files N.)rrzujson version too oldTF)AnyMappingdreturnch|jDcic]\}}|t|c}}Scc}}w)N)items process_value)rkvs H/var/www/dash_apps/app1/venv/lib/python3.12/site-packages/stanio/json.pyprocess_dictionaryrs*,-GGI 6DAqA}Q  66 6s.valc |yt|tr t|St|tr|j|j gSt|t r t|St|tr9t ttdt|dztt|St|tr|Dcgc] }t|c}Stt!|dd}d|vs d|vsd|vr t#j$|}|j&j(dvr|j+S|j&j(dk(rat#j,t#j.|jt#j.|j gd j+S|j&j(d k(r#|j1tj+St|j+S|Scc}w) N __module__numpyxarraypandasiufc)axisb) isinstanceboolintcomplexrealimagdictrtupleziprangelenmapr listgetattrtypenp asanyarraydtypekindtoliststackasarrayastype)rioriginal_module numpy_vals rr r s {#t3x#w#((###t!#&&#uCaSA.M30GHII#t*-.Q a ..d3ir:O?"  &  &MM#& ??  5 (##% % ??  3 &88RZZ 7INN9ST[]^eeg g ??  3 &##C(//1 1Y--/00 J'/s?Hdatac>tjt|S)a4 Convert a mapping of strings to data to a JSON string. Values can be any numeric type, a boolean (converted to int), or any collection compatible with :func:`numpy.asarray`, e.g a :class:`pandas.Series`. Produces a string compatible with the `Json Format for Cmdstan `__ :param data: A mapping from strings to values. This can be a dictionary or something more exotic like an :class:`xarray.Dataset`. This will be copied before type conversion, not modified )jsondumpsr)r7s rdump_stan_jsonr;;s ::(. //pathct|d5}tr tjt ||nBtj j t |D]}|j|dddy#1swYyxYw)a Dump a mapping of strings to data to a JSON file. Values can be any numeric type, a boolean (converted to int), or any collection compatible with :func:`numpy.asarray`, e.g a :class:`pandas.Series`. Produces a file compatible with the `Json Format for Cmdstan `__ :param path: File path for the created json. Will be overwritten if already in existence. :param data: A mapping from strings to values. This can be a dictionary or something more exotic like an :class:`xarray.Dataset`. This will be copied before type conversion, not modified wN)openUJSON_AVAILABLEr9dumpr JSONEncoder iterencodewrite)r=r7fdchunks rwrite_stan_jsonrHNso& dC B  II(. 3))+667I$7OP     s A)A??B)__doc__ujsonr9r$r(r __version__split uj_version ImportErrorrAtypingrrrr,strrr r;rHr<rrRs s3 0 0 6 6s ;<=JI122O  7'#s(+7S0A7 ssD0c*0s0& # WS#X%6 4 GOs ;BB