o i@sXddlmZddlmZddlmZddlmZer"ddl m Z m Z eddddZ dS)) annotations) TYPE_CHECKINGN)DEFAULT_QUERY_OPT_FLAGS) LazyFrame QueryOptFlags) optimizationslfrallow_local_scansboolrrreturnbytescCs$|}|j|j}tj||dS)a Prepare the given LazyFrame for execution on Polars Cloud. Parameters ---------- lf The LazyFrame to prepare. allow_local_scans Whether or not to allow local scans in the plan. optimizations Optimizations to enable or disable in the query optimizer. Raises ------ InvalidOperationError If the given LazyFrame is not eligible to be run on Polars Cloud. The following conditions will disqualify a LazyFrame from being eligible: - Contains a user-defined function - Scans or sinks to a local filesystem ComputeError If the given LazyFrame cannot be serialized. )r )__copy__Z_ldfZwith_optimizationsZ _pyoptflagsplrprepare_cloud_plan)rr rZpylfrJ/home/app/Keep/.python/lib/python3.10/site-packages/polars/_utils/cloud.pyr sr)rrr r rrr r ) __future__rtypingrZ polars._plrZ_plrrZpolars.lazyframe.opt_flagsrZpolarsrrrrrrrs