o 0j @sUdZddlmZddlmZddlmZddlmZm Z m Z m Z m Z m Z ddlmZddlmZed Zd ed <e e ed fe eZd ed<Gddde ddZefdddZefd ddZdS)!zPModel file paths and format resolution for inference model directory convention.) annotations)PathLike)Path)LiteralTuple TypedDictUnioncastget_args) TypeAlias)MODEL_FILE_PREFIX)paddleonnxom paddle_dyn safetensorsr LocalModelFormat.zTuple[LocalModelFormat, ...]LOCAL_MODEL_FORMATSc@s:eZdZUdZded<ded<ded<ded<ded<d S) ModelPathsz7Resolved local model files keyed by `LocalModelFormat`.zTuple[Path, Path]rrrrrrN)__name__ __module__ __qualname____doc____annotations__rrk/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/paddlex/inference/models/utils/model_paths.pyr(s  rF)total model_dirUnion[str, PathLike]model_file_prefixstrreturncCst|}i}d}||dr||d}n||dr)||d}|r?||dr?|||df|d<||drQ||d|d<||drc||d|d<|d ro|d |d <|d r{|d |d <|d r|d |d <|S)Nz.jsonz.pdmodelz .pdiparamsrz.onnxrz.omrzmodel_state.pdparamsrzinference.pdparamszmodel.safetensorsr)rexists)rr Z model_pathsZ pd_model_pathrrrget_model_paths2s,       r$Union[str, PathLike, Path] str | NonecCs6tt||}d|vr dSd|vrdSd|vrdSdS)zResolve paddle_static vs paddle_dynamic from actual model files. Returns: "paddle_static", "paddle_dynamic", or None if no paddle model files found. rZ paddle_staticrZpaddle_dynamicrN)r$r)rr pathsrrr&resolve_paddle_engine_from_model_filesOs r(N)rrr r!r"r)rr%r r!r"r&)r __future__rosrpathlibrtypingrrrrr r Ztyping_extensionsr constantsr rrrrr$r(rrrrs$