o }oi@sddlmZddlZddlZddlmZddlmZddlmZddlZddl m Z ddl m Z erJddl mZdd lmZed Zed Zed Zd Zd"ddZd#ddZd$ddZ d%d&ddZ d%d'd d!ZdS)() annotationsN)Any) TYPE_CHECKING)TypeVar) optuna_warn)ExperimentalWarning)Callable) ParamSpecFTFPCTz .. note:: Added in v{ver} as an experimental feature. The interface may change in newer versions without prior notice. See https://github.com/optuna/optuna/releases/tag/v{ver}. option_namestrreturnNonecCstd|dtdS)Nz Argument ``zF`` is an experimental feature. The interface can change in the future.)rr)r rH/home/ubuntu/.local/lib/python3.10/site-packages/optuna/_experimental.pywarn_experimental_argument s rversioncCs0t|trt|ddkrtd|ddS)N.z?Invalid version specification. Must follow `x.y.z` format but `z ` is given) isinstancerlensplit ValueError)rrrr_validate_version(s  r docstringcCsd|vr |ddSdS)N )r)rrrr_get_docstring_indent/sr name str | None.Callable[[Callable[FP, FT]], Callable[FP, FT]]ctdfdd }|S)aDecorate function as experimental. Args: version: The first version that supports the target feature. name: The name of the feature. Defaults to fully qualified name of the function, i.e. `f"{func.__module__}.{func.__qualname__}"`. Optional. funcCallable[FP, FT]rcsxjdurd_tjd}tj}jt|||_p+jdjt d fd d }|S) Nrverrargsrkwargsrr cs*tjddtdd|i|SNz" is experimental (supported from vz*). The interface can change in the future.) stacklevelwarningswarnr)r)r*)_namer%rrrwrapperKs z5experimental_func..decorator..wrapper)r)rr*rrr ) __doc___EXPERIMENTAL_NOTE_TEMPLATEformatr striptextwrapindent __module__ __qualname__ functoolswraps)r%noter8r2r!r)r1r%r decoratorAs    z$experimental_func..decoratorN)r%r&rr&rrr!r?rr>rexperimental_func3s rBCallable[[CT], CT]cr$)zDecorate class as experimental. Args: version: The first version that supports the target feature. name: The name of the feature. Defaults to the class name. Optional. clsr rcsdfdd }||S)NrDr rcst|dt|dtdfd d }t|d||jd ur'd |_tjd }t|j}|jt ||||_|S)zDecorates a class as experimental. This decorator is supposed to be applied to the experimental class. __init____name__selfrr)r*rrcsBtjdurnddtdd|g|Ri|dSr+r.)rGr)r*)_original_init_original_namer!rrr wrapped_initqszLexperimental_class..decorator..wrapper..wrapped_initNrr')rGrr)rr*rrr) getattrr;r<setattrr3r4r5r r6r7r8)rDrJr=r8r>)rHrIrr2is     z6experimental_class..decorator..wrapperrDr rr r)rDr2r>rrr?hsz%experimental_class..decoratorNrMr@rArr>rexperimental_class[s "rN)r rrr)rrrr)rrrr)N)rrr!r"rr#)rrr!r"rrC) __future__rr;r7typingrrrr/optuna._warningsroptuna.exceptionsrcollections.abcrtyping_extensionsr r r r r4rrr rBrNrrrrs.            *