Set default value of TSDataset.head method (#170)

* Set default value for head method to 5 * Update changelog
tinkoff-ai · Oct 11, 2021 · 9e9e0c4 · 9e9e0c4
1 parent 716da3a
commit 9e9e0c4
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git 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
@@ -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.