h5%SrSSKJr SSKrSSKrSSKrSSKJrJ r J r SSK J r J r SSKJr SSKJr SSKJrJrJr SS KJr SSKrS rSS KJr /S Qr0rS\S'0rS\S'S Sjr "SS\ 5r!"SS\!S9r"S!S"Sjjr#S#S$Sjjr$S#S%Sjjr%S&Sjr&S'Sjr'S(S)Sjjr(S*S+Sjjr)S,Sjr*S-Sjr+g!\a S rNf=f).aModule for serialization and deserialization of Albumentations transforms. This module provides functionality to serialize transforms to JSON or YAML format and deserialize them back. It implements the Serializable interface that allows transforms to be converted to and from dictionaries, which can then be saved to disk or transmitted over a network. This is particularly useful for saving augmentation pipelines and restoring them later with the exact same configuration. ) annotationsN)ABCABCMetaabstractmethod)MappingSequence)Enum)Path)AnyLiteralTextIO)warnTF) __version__) from_dictloadsaveto_dictzdict[str, SerializableMeta]SERIALIZABLE_REGISTRYNON_SERIALIZABLE_REGISTRYcnURS5nUS:XdURS5(dU$XS-S$)N.zalbumentations.)rfind startswith)class_fullname split_indexs [/var/www/fran/franai/venv/lib/python3.13/site-packages/albumentations/core/serialization.pyshorten_class_namer'sE &&s+Kb 9 9:K L L /+ ,,cl^\rSrSrSrSU4Sjjr\S Sj5r\S Sj5r\S Sj5r Sr U=r $) SerializableMeta3zA metaclass that is used to register classes in `SERIALIZABLE_REGISTRY` or `NON_SERIALIZABLE_REGISTRY` so they can be found later while deserializing transformation pipeline using classes full names. c>[TU]"XU/UQ70UD6nUS:waO[U;aEUR5(aU[UR 5'U$U[ UR 5'U$)N Serializable)super__new__ris_serializablerget_class_fullnamer)clsnamebasesargskwargscls_obj __class__s rr'SerializableMeta.__new__8ss'/#UDTDVD > !c&6&&((FM%g&@&@&BCKR)'*D*D*FGr cg)NFr*s rr( SerializableMeta.is_serializableAsr c[U5$N)get_shortest_class_fullnamer4s rr)#SerializableMeta.get_class_fullnameEs *3//r c0$r7r3r4s r_to_dictSerializableMeta._to_dictIs r r3) r+strr,ztuple[type, ...]r-r r.r returnr"r>boolr>r=r>dict[str, Any]) __name__ __module__ __qualname____firstlineno____doc__r' classmethodr(r)r;__static_attributes__ __classcell__)r0s@rr"r"3sI00r r"cr\rSrSr\\SSj55r\\S Sj55r\S Sj5rS S Sjjr Sr g) r%Nc[er7NotImplementedErrorr4s rr(Serializable.is_serializableO "!r c[er7rOr4s rr)Serializable.get_class_fullnameTrRr c[er7rO)selfs rto_dict_privateSerializable.to_dict_privateYs!!r cUS;aSUS3n[U5eUR5n[US .$![a< US:Xae0n[R"SUSUR R S3SS 9 NNf=f) aYTake a transform pipeline and convert it to a serializable representation that uses only standard python data types: dictionaries, lists, strings, integers, and floats. Args: self (Serializable): A transform that should be serialized. If the transform doesn't implement the `to_dict` method and `on_not_implemented_error` equals to 'raise' then `NotImplementedError` is raised. If `on_not_implemented_error` equals to 'warn' then `NotImplementedError` will be ignored but no transform parameters will be serialized. on_not_implemented_error (str): `raise` or `warn`. >rraisez(Unknown on_not_implemented_error value: z . Supported values are: 'raise' rZz2Got NotImplementedError while trying to serialize z;. Object arguments are not preserved. The transform class 'z`' needs to implement 'to_dict_private' or inherit from BasicTransform to be properly serialized. stacklevel)r transform) ValueErrorrWrPwarningsrr0rDr)rVon_not_implemented_errormsgtransform_dicts rrSerializable.to_dict]s $+< <<=UrC) rDrErFrGrIrr(r)rWrrJr3r rr%r%NsX""""""IIr r%) metaclassc$URU5$)a^Take a transform pipeline and convert it to a serializable representation that uses only standard python data types: dictionaries, lists, strings, integers, and floats. Args: transform (Serializable): A transform that should be serialized. If the transform doesn't implement the `to_dict` method and `on_not_implemented_error` equals to 'raise' then `NotImplementedError` is raised. If `on_not_implemented_error` equals to 'warn' then `NotImplementedError` will be ignored but no transform parameters will be serialized. on_not_implemented_error (str): `raise` or `warn`. )r)r^ras rrr}s   5 66r cURS5[;a?USnUcSUS3n[U5eURU5nUc[SUS35eU$g)N__class_fullname__rDz6To deserialize a non-serializable transform with name z you need to pass a dict withz Non-serializable transform with z# was not found in `nonserializable`)getrr_)r^nonserializabler+rbresult_transforms rinstantiate_nonserializablermsy}})*.GG$  "J4&PmnC @S/ !*..t4  ?vEhij j r c[5 USn[X!5nU(aU$USnUR5VVs0sHupVUS:wdM XV_M nnnSU;aUS;a[SUS3SS9 S US'[[ U5nS U;a#US V s/sHn [ SU 0US 9PM sn US 'U"S 0UD6$s snnfs sn f) aArgs: transform_dict: A dictionary with serialized transform pipeline. nonserializable (dict): A dictionary that contains non-serializable transforms. This dictionary is required when you are restoring a pipeline that contains non-serializable transforms. Keys in that dictionary should be named same as `name` arguments in respective transforms from a serialized pipeline. r^rip)Compose Sequentialz Transform z; has no 'p' parameter in serialized data, defaulting to 0.5r[r\g? transformsrkr3)register_additional_transformsrmitemsrrrr) rcrkr^lmbdr+kvr-r*ts rrrs#${+I &y BD  ) *D&__. L.TQ!7K2KDAD.D L $4'@@ z$Z [hijS  24 8 9Ctdhiudvwdv_`ia(8/Zdvw\ ;; Mxs B7 B7B=c.US;a[SUS35eg)N>jsonyamlzUnknown data_format z*. Supported formats are: 'json' and 'yaml')r_) data_formats rcheck_data_formatr~s&**/ }!$$==#x  C)=)=+./3aq!3//"3--399636>/s B.3B4c.[U5 URUS9n[U5n[U[[ 45(at[ U5R S5nUS:Xa.[(d Sn[U5e[R"XESS9 OUS:Xa[R"XE5 SSS5 gUS:Xa.[(d Sn[U5e[R"XASS9 gUS:Xa[R"XAS S 9 gg!,(df  g=f) aSerialize a transform pipeline and save it to either a file specified by a path or a file-like object in either JSON or YAML format. Args: transform (Serializable): The transform pipeline to serialize. filepath_or_buffer (Union[str, Path, TextIO]): The file path or file-like object to write the serialized data to. If a string is provided, it is interpreted as a path to a file. If a file-like object is provided, the serialized data will be written to it directly. data_format (str): The format to serialize the data in. Valid options are 'json' and 'yaml'. Defaults to 'json'. on_not_implemented_error (str): Determines the behavior if a transform does not implement the `to_dict` method. If set to 'raise', a `NotImplementedError` is raised. If set to 'warn', the exception is ignored, and no transform arguments are saved. Defaults to 'raise'. Raises: ValueError: If `data_format` is 'yaml' but PyYAML is not installed. )rawr|z A  A cNURSUR3n[U5$)a\The function `get_shortest_class_fullname` takes a class object as input and returns its shortened full name. :param cls: The parameter `cls` is of type `Type[BasicCompose]`, which means it expects a class that is a subclass of `BasicCompose` :type cls: Type[BasicCompose] :return: a string, which is the shortened version of the full class name. r)rErDr)r*rs rr8r8;s('q7N n --r )rr=r>r=re)r^r%rar=r>rCr7)r^rCrkdict[str, Any] | Noner>Serializable | None)rcrCrkrr>r)r}Literal['json', 'yaml']r>None)rr r>r )r{rZ) r^r%rstr | Path | TextIOr}rrazLiteral['raise', 'warn']r>r)r{N)rrr}rrkrr>object)r>r)r*z type[Any]r>r=),rH __future__rimportlib.utilrr{r`abcrrrcollections.abcrrenumr pathlibr typingr r r rr|rrrr__all__r__annotations__rrr"r%rrmrr~rrrrtr8r3r rrst# ,,-''N ' 268279;6; -w6,I-,I^ 7".2*&.2"*>i 7,29@ -@-@+-@)-@7 -@  -@d,2-1-F+-F(-F+-F -F`  .C NsCCC