3 [¾ hŒã@sddZddlZddlmZeejjddƒƒeejjddƒƒdd d œZd d „Z dd d„Z edd„ƒZ dS)z8Global configuration state and functions for management éN)ÚcontextmanagerZSKLEARN_ASSUME_FINITEFZSKLEARN_WORKING_MEMORYiTÚtext)Ú assume_finiteÚworking_memoryÚprint_changed_onlyÚdisplaycCstjƒS)a]Retrieve current values for configuration set by :func:`set_config` Returns ------- config : dict Keys are parameter names that can be passed to :func:`set_config`. See Also -------- config_context : Context manager for global scikit-learn configuration. set_config : Set global scikit-learn configuration. )Ú_global_configÚcopy©r r ú7/tmp/pip-build-zwgx3nbq/scikit-learn/sklearn/_config.pyÚ get_configs r cCsD|dk r|td<|dk r |td<|dk r0|td<|dk r@|td<dS)a>Set global scikit-learn configuration .. versionadded:: 0.19 Parameters ---------- assume_finite : bool, default=None If True, validation for finiteness will be skipped, saving time, but leading to potential crashes. If False, validation for finiteness will be performed, avoiding error. Global default: False. .. versionadded:: 0.19 working_memory : int, default=None If set, scikit-learn will attempt to limit the size of temporary arrays to this number of MiB (per job when parallelised), often saving both computation time and memory on expensive operations that can be performed in chunks. Global default: 1024. .. versionadded:: 0.20 print_changed_only : bool, default=None If True, only the parameters that were set to non-default values will be printed when printing an estimator. For example, ``print(SVC())`` while True will only print 'SVC()' while the default behaviour would be to print 'SVC(C=1.0, cache_size=200, ...)' with all the non-changed parameters. .. versionadded:: 0.21 display : {'text', 'diagram'}, default=None If 'diagram', estimators will be displayed as a diagram in a Jupyter lab or notebook context. If 'text', estimators will be displayed as text. Default is 'text'. .. versionadded:: 0.23 See Also -------- config_context : Context manager for global scikit-learn configuration. get_config : Retrieve current values of the global configuration. Nrrrr)r)rrrrr r r Ú set_configs-r c ks0tƒjƒ}tf|Žz dVWdtf|ŽXdS)a¢Context manager for global scikit-learn configuration Parameters ---------- assume_finite : bool, default=False If True, validation for finiteness will be skipped, saving time, but leading to potential crashes. If False, validation for finiteness will be performed, avoiding error. Global default: False. working_memory : int, default=1024 If set, scikit-learn will attempt to limit the size of temporary arrays to this number of MiB (per job when parallelised), often saving both computation time and memory on expensive operations that can be performed in chunks. Global default: 1024. print_changed_only : bool, default=True If True, only the parameters that were set to non-default values will be printed when printing an estimator. For example, ``print(SVC())`` while True will only print 'SVC()', but would print 'SVC(C=1.0, cache_size=200, ...)' with all the non-changed parameters when False. Default is True. .. versionchanged:: 0.23 Default changed from False to True. display : {'text', 'diagram'}, default='text' If 'diagram', estimators will be displayed as a diagram in a Jupyter lab or notebook context. If 'text', estimators will be displayed as text. Default is 'text'. .. versionadded:: 0.23 Notes ----- All settings, not just those presently modified, will be returned to their previous values when the context manager is exited. This is not thread-safe. Examples -------- >>> import sklearn >>> from sklearn.utils.validation import assert_all_finite >>> with sklearn.config_context(assume_finite=True): ... assert_all_finite([float('nan')]) >>> with sklearn.config_context(assume_finite=True): ... with sklearn.config_context(assume_finite=False): ... assert_all_finite([float('nan')]) Traceback (most recent call last): ... ValueError: Input contains NaN, ... See Also -------- set_config : Set global scikit-learn configuration. get_config : Retrieve current values of the global configuration. N)r r r )Z new_configZ old_configr r r Úconfig_contextUs ;   r)NNNN) Ú__doc__ÚosÚ contextlibrÚboolÚenvironÚgetÚintrr r rr r r r Ús  6