From 9e9e0c45c3454126edacefc601862c8505fa474d Mon Sep 17 00:00:00 2001
From: Mr-Geekman <36005824+Mr-Geekman@users.noreply.github.com>
Date: Mon, 11 Oct 2021 12:17:51 +0300
Subject: [PATCH] Set default value of `TSDataset.head` method (#170)

* Set default value for head method to 5

* Update changelog
---
 CHANGELOG.md               | 1 +
 etna/datasets/tsdataset.py | 2 +-
 2 files changed, 2 insertions(+), 1 deletion(-)

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.