polars.Config.set_engine_affinity

classmethod Config.set_engine_affinity(
    engine: EngineType | None = None,
) → type[Config]

Задать движок, используемый по умолчанию.

Параметры:

engine{None, ‘auto’, ‘in-memory’, ‘streaming’, ‘gpu’}

Движок выполнения по умолчанию, который Polars попытается использовать при вызове .collect(). Однако выполнение запроса с указанным движком не гарантируется.

Для настройки значения по умолчанию также можно использовать объект Engine. Объекты движков привязаны к процессу и не включаются в Config.save(); загрузка состояния, содержащего POLARS_ENGINE_AFFINITY, заменяет привязку объекта.

Вызывает исключения:

ValueError: если движок не распознан.

Примеры

>>> pl.Config.set_engine_affinity("streaming")  
>>> lf = pl.LazyFrame({"v": [1, 2, 3], "v2": [4, 5, 6]})  
>>> lf.max().collect()  
shape: (3, 2)
┌─────┬─────┐
│ v   ┆ v2  │
│ --- ┆ --- │
│ i64 ┆ i64 │
╞═════╪═════╡
│ 1   ┆ 4   │
│ 2   ┆ 5   │
│ 3   ┆ 6   │
└─────┴─────┘
>>> pl.Config.set_engine_affinity("gpu")  
>>> lf.max().collect()  
shape: (3, 2)
┌─────┬─────┐
│ v   ┆ v2  │
│ --- ┆ --- │
│ i64 ┆ i64 │
╞═════╪═════╡
│ 1   ┆ 4   │
│ 2   ┆ 5   │
│ 3   ┆ 6   │
└─────┴─────┘

Задать настроенный движок по умолчанию:

>>> pl.Config.set_engine_affinity(
...     pl.GPUEngine(device=1, raise_on_fail=True)
... )