From 7b8a5f75ea915d887f35f14b00b8f190930f29c6 Mon Sep 17 00:00:00 2001
From: Egor Baturin <egoriyaa@github.com>
Date: Fri, 8 Nov 2024 14:44:31 +0300
Subject: [PATCH] make attributes private

---
 etna/transforms/encoders/mean_segment_encoder.py | 15 +++++++--------
 1 file changed, 7 insertions(+), 8 deletions(-)

diff --git a/etna/transforms/encoders/mean_segment_encoder.py b/etna/transforms/encoders/mean_segment_encoder.py
index 02c9c6010..c5037e077 100644
--- a/etna/transforms/encoders/mean_segment_encoder.py
+++ b/etna/transforms/encoders/mean_segment_encoder.py
@@ -10,14 +10,13 @@
 class MeanSegmentEncoderTransform(IrreversibleTransform):
     """Makes expanding mean target encoding of the segment. Creates column 'segment_mean'."""
 
-    idx = pd.IndexSlice
-    segment_column = "segment_column"
+    _segment_column = "segment_column"
     out_column = "segment_mean"
 
     def __init__(self):
         super().__init__(required_features=["target"])
-        self.mean_encoder = MeanEncoderTransform(
-            in_column=self.segment_column, mode="per-segment", out_column=self.out_column, smoothing=0
+        self._mean_encoder = MeanEncoderTransform(
+            in_column=self._segment_column, mode="per-segment", out_column=self.out_column, smoothing=0
         )
 
     def _add_segment_column(self, df):
@@ -25,7 +24,7 @@ def _add_segment_column(self, df):
         flatten_segments = np.repeat(segments.values[np.newaxis, :], len(df), axis=0)
         segment_values = pd.DataFrame(
             data=flatten_segments,
-            columns=pd.MultiIndex.from_product([segments, [self.segment_column]]),
+            columns=pd.MultiIndex.from_product([segments, [self._segment_column]]),
             index=df.index,
         )
         df = pd.concat([df, segment_values], axis=1).sort_index(axis=1)
@@ -46,7 +45,7 @@ def _fit(self, df: pd.DataFrame) -> "MeanSegmentEncoderTransform":
             Fitted transform
         """
         df = self._add_segment_column(df)
-        self.mean_encoder._fit(df)
+        self._mean_encoder._fit(df)
         return self
 
     def _transform(self, df: pd.DataFrame) -> pd.DataFrame:
@@ -71,8 +70,8 @@ def _transform(self, df: pd.DataFrame) -> pd.DataFrame:
             If there are segments that weren't present during training.
         """
         df = self._add_segment_column(df)
-        df_transformed = self.mean_encoder._transform(df)
-        df_transformed = df_transformed.drop(columns=[self.segment_column], level="feature")
+        df_transformed = self._mean_encoder._transform(df)
+        df_transformed = df_transformed.drop(columns=[self._segment_column], level="feature")
         return df_transformed
 
     def get_regressors_info(self) -> List[str]: