diff --git a/CHANGELOG.md b/CHANGELOG.md index ff7b5c463..74e14d794 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -36,6 +36,7 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0 - Defined random state for every test cases ([#155](https://github.com/tinkoff-ai/etna-ts/pull/155)) ### Fixed +- Set default value of `TSDataset.head` method ([#170](https://github.com/tinkoff-ai/etna-ts/pull/170)) ## [1.1.3] - 2021-10-08 ### Fixed diff --git a/etna/datasets/tsdataset.py b/etna/datasets/tsdataset.py index e938ec198..982cbd44d 100644 --- a/etna/datasets/tsdataset.py +++ b/etna/datasets/tsdataset.py @@ -576,7 +576,7 @@ def isnull(self) -> pd.DataFrame: """ return self.df.isnull() - def head(self, n_rows: Optional[int] = None) -> pd.DataFrame: + def head(self, n_rows: int = 5) -> pd.DataFrame: """Return the first `n` rows. Mimics pandas method.