IZj |ddlmZddlZddlZddlmZddlmZddlm Z ddl m Z ddl m Z mZmZmZmZdZdZdZdZdZdZdZdZGd d e Zd+dZe r!ddlZddlZddlZddlZddl Z ddl!Z!ddl"Z"ddl#Z#nped\ZZed\ZZed\ZZed\Z"Zed\ZZed\Z!Zed\Z Zed\Z#Zedd,dZ$ddd-d"Z%ddd-d#Z&ddd-d$Z'ddd-d%Z(ddd-d&Z)ddd-d'Z*ddd-d(Z+ddd-d)Z,gd*Z-dS).) annotationsN) lru_cache) import_module) find_spec) ModuleType) TYPE_CHECKINGAnyClassVarHashablecastTcXeZdZUdZdZddddddd d Zd ed <dfd ZddZddZ xZ S) _LazyModulea/ Module that can act both as a lazy-loader and as a proxy. Notes ----- We do NOT register this module with `sys.modules` so as not to cause confusion in the global environment. This way we have a valid proxy module for our own use, but it lives _exclusively_ within lance. Tznp.zpd.zpl.ztorch.ztf.zray.zlance.)numpypandaspolarstorch tensorflowraylancezClassVar[dict[str, str]]_mod_pfx module_namestrmodule_availableboolreturnNonec||_||_t|_t |dS)a Initialise lazy-loading proxy module. Parameters ---------- module_name : str the name of the module to lazy-load (if available). module_available : bool indicate if the referenced module is actually available (we will proxy it in both cases, but raise a helpful error when invoked if it doesn't exist). N)_module_available _module_nameglobals_globalssuper__init__)selfrr __class__s `/Users/jameslopez/projects/MentorCore/.venv/lib/python3.11/site-packages/lancedb/dependencies.pyr#z_LazyModule.__init__<s>&"2'   %%%%%rct|j}||j|j<|j|j|S)N)r__name__r!r__dict__update)r$modules r&_importz_LazyModule._importTs=t}--+1 d'( V_--- r'attrr cd|dkrt|jd||jr$|}t ||S|dkr|jSt jd|r|dkrdS|j|jd}t||d|jdd) N __wrapped__z object has no attribute r)z ^__\w+__$ __version__z requires z module to be installed) AttributeErrorrrr-getattrrematchrgetModuleNotFoundError)r$r.r,pfxs r& __getattr__z_LazyModule.__getattr__[s =  $IIII   ! \\^^F64(( (Z  $ $ XlD ) ) dm.C.C4-##D$5r::C%TTT(9TTT r')rrrrrr)rr)r.r rr ) r) __module__ __qualname____doc____lazy__r__annotations__r#r-r: __classcell__)r%s@r&rr$s  H**H&&&&&&0r'rrrrtuple[ModuleType, bool]c|tjvrtj|dfS t|}|dup|jdu }n#t$rd}YnwxYwt |||fS)a Lazy import the given module; avoids up-front import costs. Parameters ---------- module_name : str name of the module to import, eg: "polars". Notes ----- If the requested module is not available (eg: has not been installed), a proxy module is created in its place, which raises an exception on any attribute access. This allows for import and use as normal, without requiring explicit guard conditions - if the module is never used, no exception occurs; if it is, then a helpful exception is raised. Returns ------- tuple of (Module, bool) A lazy-loading module and a boolean indicating if the requested/underlying module exists (if not, the returned module is a proxy). TNF)rr)sysmodulesrloaderr8r)r module_specrs r& _lazy_importrGxs2ck!!{;'--! ,, +t 3 Q{7IT7QR !!! ! #-      sA AArrrrdatasetsrrr)maxsizeclstypetype_rc tfd|DS#t$rYdSwxYw)Nc3@K|]}dt|vVdS).N)r).0orLs r& z_might_be..s4<>) check_typeobjr rWcxto3ttt|rt |n|dS)Nr)_NUMPY_AVAILABLErVr r rKrXrWs r&_check_for_numpyr\9    XJ7tCyyyC88'!!r'cxto3ttt|rt |n|dS)Nr)_PANDAS_AVAILABLErVr r rKr[s r&_check_for_pandasr`9   XJ7tCyyyC88(""r'cxto3ttt|rt |n|dS)Nr)_POLARS_AVAILABLErVr r rKr[s r&_check_for_polarsrdrar'cxto3ttt|rt |n|dS)Nr)_TORCH_AVAILABLErVr r rKr[s r&_check_for_torchrgr]r'cxto3ttt|rt |n|dS)NrH)_HUGGING_FACE_AVAILABLErVr r rKr[s r&_check_for_hugging_facerjs9 " y XJ7tCyyyC88*((r'cxto3ttt|rt |n|dS)Nr)_TENSORFLOW_AVAILABLErVr r rKr[s r&_check_for_tensorflowrms9  Y XJ7tCyyyC88,&&r'cxto3ttt|rt |n|dS)Nr)_RAY_AVAILABLErVr r rKr[s r&_check_for_rayrps9  i XJ7tCyyyC88%r'cxto3ttt|rt |n|dS)Nr)_LANCE_AVAILABLErVr r rKr[s r&_check_for_lancersr]r')rHrrrrrrrrjr\r`rdrmrgrprsrrZr_rcrfrirlrorr)rrrrA)rJrKrLrrr)rXr rWrrr). __future__rr5rC functoolsr importlibrimportlib.utilrtypesrtypingrr r r r rZr_rcrfrirlrorrrrGrHrrrrrrrrVr\r`rdrgrjrmrprs__all__r'r&r|sp #""""" ######$$$$$$??????????????QQQQQ*QQQh++++\4OOOLLLMMMMMMJJJLLLLLLL+l733E  , X 6 6F  , X 6 6F *l733E (4 Z(@(@%H%(4 \(B(B%J%&,u--C*l733E  46: 7; 7; 6: =A ;? 48 6:    r'