U Cv f*+@sUddlmZddlZddlmZmZmZddlmZer|zddl m Z Wne k rbe Z YnXedZ GdddeZd d d d d dddddddddddddddZded<ed ddddgZeddgZdddddZejeeejedd d!d d"d d#d d$d d%d d&d ed'd d(d d)d d*d d+Zd,d-d.d/Zd0d1Zeed2Zd3ddd4d5d6Zddd7d8d9ZGd:d;d;Zd) annotationsN) TYPE_CHECKINGLiteral TypedDict) FrozenDict)Colormap)arithmetic_joincmap_divergentcmap_sequentialdisplay_max_rowsdisplay_values_threshold display_style display_widthdisplay_expand_attrsdisplay_expand_coordsdisplay_expand_data_varsdisplay_expand_datadisplay_expand_indexesdisplay_default_indexesenable_cftimeindexfile_cache_maxsize keep_attrswarn_for_unclosed_filesuse_bottleneckuse_floxc@seZdZUded<ded<ded<ded<ded<d ed <ded <d ed <d ed<d ed<d ed<d ed<d ed<ded<ded<d ed<ded<ded<ded<dS) T_Optionsz5Literal[('inner', 'outer', 'left', 'right', 'exact')]rzstr | Colormapr r intr r zLiteral[('text', 'html')]r rz!Literal[('default', True, False)]rrrrrrboolrrrrrrN)__name__ __module__ __qualname____annotations__r"r"7/tmp/pip-unpacked-wheel-h316xyqg/xarray/core/options.pyr#s& rinnerZRdBu_rZviridis htmlPdefaultFTOPTIONSouterleftrightexacttextrr)valuereturncCst|to|dkS)Nr) isinstancerr1r"r"r#_positive_integerSsr5cCs|dkSN)TFr)r"choicer"r"r#]r9cCs|dkSr6r"r7r"r"r#r9^r:cCs|dkSr6r"r7r"r"r#r9_r:cCs|dkSr6r"r7r"r"r#r9`r:cCs|dkSr6r"r7r"r"r#r9ar:cCs|dkSr6r"r7r"r"r#r9br:cCs t|tSNr3rr4r"r"r#r9cr:cCs|dkSr6r"r7r"r"r#r9er:cCs t|tSr;r<r4r"r"r#r9fr:cCs t|tSr;r<r4r"r"r#r9gr:cCs t|tSr;r<r4r"r"r#r9hr:)rr r r rrrrrrrrrrrrrNone)r2cCsddlm}||_dS)Nr) FILE_CACHE)Zxarray.backends.file_managerr>maxsize)r1r>r"r"r#_set_file_cache_maxsizels r@cCstdtdS)Nz_The enable_cftimeindex option is now a no-op and will be removed in a future version of xarray.)warningswarn FutureWarning)rr"r"r##_warn_on_setting_enable_cftimeindexrsrD)rrOptions)optionr)r2cCs6t|}|dkr|St|tr"|Std|ddS)Nr)zThe global option z) must be one of True, False or 'default'.)r+r3r ValueError)rFr)Z global_choicer"r"r#_get_boolean_with_defaults  rH)r)r2cCs td|S)Nr)rH)r)r"r"r#_get_keep_attrssrIc@s0eZdZdZddZddZddZdd Zd S) set_optionsa Set options for xarray in a controlled context. Parameters ---------- arithmetic_join : {"inner", "outer", "left", "right", "exact"}, default: "inner" DataArray/Dataset alignment in binary operations: - "outer": use the union of object indexes - "inner": use the intersection of object indexes - "left": use indexes from the first object with each dimension - "right": use indexes from the last object with each dimension - "exact": instead of aligning, raise `ValueError` when indexes to be aligned are not equal - "override": if indexes are of same size, rewrite indexes to be those of the first object with that dimension. Indexes for the same dimension must have the same size in all objects. cmap_divergent : str or matplotlib.colors.Colormap, default: "RdBu_r" Colormap to use for divergent data plots. If string, must be matplotlib built-in colormap. Can also be a Colormap object (e.g. mpl.cm.magma) cmap_sequential : str or matplotlib.colors.Colormap, default: "viridis" Colormap to use for nondivergent data plots. If string, must be matplotlib built-in colormap. Can also be a Colormap object (e.g. mpl.cm.magma) display_expand_attrs : {"default", True, False} Whether to expand the attributes section for display of ``DataArray`` or ``Dataset`` objects. Can be * ``True`` : to always expand attrs * ``False`` : to always collapse attrs * ``default`` : to expand unless over a pre-defined limit display_expand_coords : {"default", True, False} Whether to expand the coordinates section for display of ``DataArray`` or ``Dataset`` objects. Can be * ``True`` : to always expand coordinates * ``False`` : to always collapse coordinates * ``default`` : to expand unless over a pre-defined limit display_expand_data : {"default", True, False} Whether to expand the data section for display of ``DataArray`` objects. Can be * ``True`` : to always expand data * ``False`` : to always collapse data * ``default`` : to expand unless over a pre-defined limit display_expand_data_vars : {"default", True, False} Whether to expand the data variables section for display of ``Dataset`` objects. Can be * ``True`` : to always expand data variables * ``False`` : to always collapse data variables * ``default`` : to expand unless over a pre-defined limit display_expand_indexes : {"default", True, False} Whether to expand the indexes section for display of ``DataArray`` or ``Dataset``. Can be * ``True`` : to always expand indexes * ``False`` : to always collapse indexes * ``default`` : to expand unless over a pre-defined limit (always collapse for html style) display_max_rows : int, default: 12 Maximum display rows. display_values_threshold : int, default: 200 Total number of array elements which trigger summarization rather than full repr for variable data views (numpy arrays). display_style : {"text", "html"}, default: "html" Display style to use in jupyter for xarray objects. display_width : int, default: 80 Maximum display width for ``repr`` on xarray objects. file_cache_maxsize : int, default: 128 Maximum number of open files to hold in xarray's global least-recently-usage cached. This should be smaller than your system's per-process file descriptor limit, e.g., ``ulimit -n`` on Linux. keep_attrs : {"default", True, False} Whether to keep attributes on xarray Datasets/dataarrays after operations. Can be * ``True`` : to always keep attrs * ``False`` : to always discard attrs * ``default`` : to use original logic that attrs should only be kept in unambiguous circumstances use_bottleneck : bool, default: True Whether to use ``bottleneck`` to accelerate 1D reductions and 1D rolling reduction operations. use_flox : bool, default: True Whether to use ``numpy_groupies`` and `flox`` to accelerate groupby and resampling reductions. warn_for_unclosed_files : bool, default: False Whether or not to issue a warning when unclosed files are deallocated. This is mostly useful for debugging. Examples -------- It is possible to use ``set_options`` either as a context manager: >>> ds = xr.Dataset({"x": np.arange(1000)}) >>> with xr.set_options(display_width=40): ... print(ds) ... Dimensions: (x: 1000) Coordinates: * x (x) int64 0 1 2 ... 998 999 Data variables: *empty* Or to set global options: >>> xr.set_options(display_width=80) # doctest: +ELLIPSIS cKsi|_|D]\}}|tkr6td|dtt|tkrt||s|dkr^dt}n|dkrrdt}nd}td|d|d |t||j|<q||dS) Nzargument name z$ is not in the set of valid options rzExpected one of r zoption z given an invalid value: z. ) olditemsr+rGset _VALIDATORS _JOIN_OPTIONS_DISPLAY_OPTIONS _apply_update)selfkwargskvexpectedr"r"r#__init__s"  zset_options.__init__cCs4|D]\}}|tkrt||qt|dSr;)rM_SETTERSr+update)rSZ options_dictrUrVr"r"r#rRszset_options._apply_updatecCsdSr;r")rSr"r"r# __enter__szset_options.__enter__cCs||jdSr;)rRrL)rStyper1 tracebackr"r"r#__exit__!szset_options.__exit__N)rrr __doc__rXrRr[r^r"r"r"r#rJs rrJcCsttS)zO Get options for xarray. See Also ---------- set_options )rr+r"r"r"r# get_options%s r`) __future__rrAtypingrrrZxarray.core.utilsrZmatplotlib.colorsr ImportErrorstrrErr+r! frozensetrPrQr5 __contains__rOr@rDrYrHrIrJr`r"r"r"r#s