Skip to content

Commit

Permalink
Fix PR 4
Browse files Browse the repository at this point in the history
  • Loading branch information
malodetz committed Aug 7, 2023
1 parent 3dfac4b commit ae6c86b
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion etna/transforms/timestamp/holiday.py
Original file line number Diff line number Diff line change
Expand Up @@ -89,8 +89,8 @@ def _transform(self, df: pd.DataFrame) -> pd.DataFrame:
raise ValueError("Frequency of data should be no more than daily.")

cols = df.columns.get_level_values("segment").unique()

out_column = self._get_column_name()

if self._mode is HolidayTransformMode.category:
encoded_matrix = np.array(
[self.holidays[x] if x in self.holidays else self._no_holiday_name for x in df.index]
Expand Down

0 comments on commit ae6c86b

Please sign in to comment.