diff --git a/CHANGELOG.md b/CHANGELOG.md index dc54b2758..4058507b4 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -35,6 +35,7 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0 - - ### Fixed +- Type hints for external users by [PEP 561](https://mypy.readthedocs.io/en/stable/running_mypy.html#missing-library-stubs-or-py-typed-marker) ([#868](https://github.com/tinkoff-ai/etna/pull/868)) - Type hints for `Pipeline.model` match `models.nn`([#768](https://github.com/tinkoff-ai/etna/pull/840)) - - diff --git a/etna/libs/__init__.py b/etna/libs/__init__.py new file mode 100644 index 000000000..e69de29bb diff --git a/etna/libs/pmdarima_utils/arima.pyi b/etna/libs/pmdarima_utils/arima.pyi new file mode 100644 index 000000000..22309ef77 --- /dev/null +++ b/etna/libs/pmdarima_utils/arima.pyi @@ -0,0 +1,18 @@ +from typing import Any + +from numpy import ndarray + +def ARMAtoMA( + ar: ndarray, + ma: ndarray, + max_deg: int, + ) -> ndarray: ... + +def seasonal_prediction_with_confidence( + arima_res: Any, + start: Any, + end: Any, + X: Any, + alpha: Any, + **kwargs: Any, + ) -> Any: ... diff --git a/etna/libs/tsfresh/distribution.pyi b/etna/libs/tsfresh/distribution.pyi new file mode 100644 index 000000000..7eeb505f7 --- /dev/null +++ b/etna/libs/tsfresh/distribution.pyi @@ -0,0 +1 @@ +def initialize_warnings_in_workers(show_warnings: bool) -> None: ... diff --git a/etna/libs/tsfresh/relevance.pyi b/etna/libs/tsfresh/relevance.pyi new file mode 100644 index 000000000..79568ade1 --- /dev/null +++ b/etna/libs/tsfresh/relevance.pyi @@ -0,0 +1,25 @@ +from typing import List +from typing import Optional + +import pandas as pd + +def calculate_relevance_table( + X: pd.DataFrame, + y: pd.Series, + ml_task: str = ..., + multiclass: bool = ..., + n_significant: int = ..., + n_jobs: int = ..., + show_warnings: bool = ..., + chunksize: Optional[int] = ..., + test_for_binary_target_binary_feature: str = ..., + test_for_binary_target_real_feature: str = ..., + test_for_real_target_binary_feature: str = ..., + test_for_real_target_real_feature: str = ..., + fdr_level: float = ..., + hypotheses_independent: bool = ..., + ) -> pd.DataFrame: ... + +def infer_ml_task(y: pd.Series) -> str: ... +def combine_relevance_tables(relevance_tables: List[pd.DataFrame]) -> pd.DataFrame: ... +def get_feature_type(feature_column: pd.Series) -> str: ... diff --git a/etna/libs/tsfresh/significance_tests.pyi b/etna/libs/tsfresh/significance_tests.pyi new file mode 100644 index 000000000..bf589824a --- /dev/null +++ b/etna/libs/tsfresh/significance_tests.pyi @@ -0,0 +1,6 @@ +import pandas as pd + +def target_binary_feature_binary_test(x: pd.Series, y: pd.Series) -> float: ... +def target_binary_feature_real_test(x: pd.Series, y: pd.Series, test: str) -> float: ... +def target_real_feature_binary_test(x: pd.Series, y: pd.Series) -> float: ... +def target_real_feature_real_test(x: pd.Series, y: pd.Series) -> float: ... diff --git a/etna/py.typed b/etna/py.typed new file mode 100644 index 000000000..e69de29bb