o Uݢg@s\dZddlmZddlZddlZddlZdejdgZddZ ddZ e ej j ej _ dS) zA wrapper to avoid importing tables directly. We need to monkey patch it on the fly to avoid issues with it changing thread usage under the hood. See CELLRANGER-2936 ) annotationsNFcCstdsFtdd7<z%ddl}|}|td<|tj_dtd<tjdtddWtdStjdtddYtdStdS) zUMemoized function to query martian for thread usage,. assuming it's loaded. rNTz%Martian runtime detected. Try Num. =  z1Warning: No martian runtime detected. Try Num. = ) _CACHED_MARTIAN_STATUSmartianget_threads_allocationtables parametersMAX_NUMEXPR_THREADSsysstdoutwrite)r n_threadsrh/oak/stanford/groups/akundaje/marinovg/programs/cellranger-9.0.1/lib/python/cellranger/wrapped_tables.py_get_n_threadssrcsfdd}|S)zuA decorator around the initialization function. to make sure it doesn't set the threads above what we want. cs$t}||d<||d<|i|S)Nr ZMAX_BLOSC_THREADS)r)argskwargsrfunctionrrwrapper1sz)file_constructor_wrapper..wrapperr)rrrrrfile_constructor_wrapper+s r) __doc__ __future__rr Znumexprr utilsZdetect_number_of_coresrrrFile__init__rrrrs