tinkoff-ai / etna

ETNA – Time-Series Library

Home Page:https://etna.tinkoff.ru

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Add processing of `PoolGenerator` class for `pool` parameter in `Auto.__init__`

Mr-Geekman opened this issue Β· comments

πŸš€ Feature Request

Currently, the type hint for pool is Union[Pool, List[BasePipeline]]. It is strange that PoolGenerator isn't possible.

Proposal

Make type hint: Union[Pool, PoolGenerator, List[BasePipeline]]. Add processing for PoolGenerator case in a code.

Test cases

  • Add new test on running with custom PoolGenerator

Additional context

Blocked by #1269.

It was decided to keep PoolGenerator class private and don't expose its API, because it isn't user-friendly.