tinkoff-ai / etna

ETNA – Time-Series Library

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

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Add possibility to disable logging of `etna.models.SARIMAXModel`

Mr-Geekman opened this issue Β· comments

πŸš€ Feature Request

Currently model etna.models.SARIMAXModel produces a lot of logs during fitting. We should give a possibility to disable it.

Proposal

Add parameter fit_params, that should accept Optional[Dict[str, Any]] = None and pass given parameters into fit in etna.models.sarimax._SARIMAXAdapter._get_fit_results.

Add in documentation that this parameter can be used to suppress logging. It can be done like this.

I don't think that this change of the signature can lead to breaking change.

Test cases

We should make sure that given parameters are passed into the model.

Additional context

No response