change variable name

RektPunk · Sep 13, 2024 · 28123e1 · 28123e1
1 parent 439c821
commit 28123e1
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/rektgbm/dataset.py b/rektgbm/dataset.py
@@ -89,7 +89,7 @@ def fit_transform_label(self) -> RektLabelEncoder:
             return self.label_encoder
         self.label_encoder = RektLabelEncoder()
         self.label = self.label_encoder.fit_transform_label(series=self.label)
-        self._is_transformed = True
+        self._is_label_transformed = True
         return self.label_encoder
 
     def transform_label(self, label_encoder: RektLabelEncoder) -> None:
@@ -152,4 +152,4 @@ def __check_label_available(self) -> None:
 
     @property
     def __is_label_transformed(self) -> bool:
-        return getattr(self, "_is_transformed", False)
+        return getattr(self, "_is_label_transformed", False)
diff --git a/tests/test_dataset.py b/tests/test_dataset.py
@@ -38,7 +38,7 @@ def test_rektdataset_fit_transform_label():
     dataset = RektDataset(data=classificatation_data, label=classification_label)
     label_encoder = dataset.fit_transform_label()
     assert isinstance(label_encoder, RektLabelEncoder)
-    assert dataset._is_transformed is True
+    assert dataset._is_label_transformed is True
 
 
 # Test transform_label