B T\@sldZddlmZmZmZddlZddlmZddlm Z e j Z e Z ddZ d d d ZGd d d eZdS)z$ Control global computation context )absolute_importdivisionprint_functionN)partial)configcOs tddS)z) Deprecated: see dask.config.set instead zThe dask.set_options function has been deprecated. Please use dask.config.set instead Before: with dask.set_options(foo='bar'): ... After: with dask.config.set(foo='bar'): ...N) TypeError)argskwargsr +lib/python3.7/site-packages/dask/context.py set_optionssr cCs$|dkrtt||dSt|||dS)a Allow function to be taken over by globals This modifies a method so that occurrences of it may be taken over by functions registered in the global options. Can be used as a decorator or a function. Parameters ---------- default : callable The default callable to use. key : str Key under which we register this function in the global parameters falsey : callable, None, optional A function to use if the option is falsey. If not provided, the default is used instead. Examples -------- >>> import dask >>> class Foo(object): ... @globalmethod(key='bar', falsey=lambda: 3) ... def bar(): ... return 1 >>> f = Foo() >>> f.bar() 1 >>> with dask.config.set(bar=lambda: 2): ... print(f.bar()) 2 >>> with dask.config.set(bar=False): ... print(f.bar()) 3 N)keyfalsey)defaultrr)r globalmethod GlobalMethod)rrrr r r rs"rc@s eZdZdddZdddZdS)rNcCs||_t||_||_dS)N)_defaultrZ normalize_key_key_falsey)selfrrrr r r __init__Bs zGlobalMethod.__init__cCs4|jtkr.t|jrt|jS|jdk r.|jS|jS)N)r_globalsrr)rinstanceownerr r r __get__Gs     zGlobalMethod.__get__)N)N)__name__ __module__ __qualname__rrr r r r rAs r)NNN)__doc__Z __future__rrrZ threading functoolsrrrZlocalZ thread_stater robjectrr r r r s   '