[Chore] change help keyword in cli and type hint (#26)

RektPunk · Sep 10, 2024 · 8f4a92e · 8f4a92e
1 parent e9c4293
commit 8f4a92e
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/rektgbm/cli.py b/rektgbm/cli.py
@@ -46,7 +46,7 @@ def main(
         typer.Argument(help="Path to save the prediction results."),
     ] = "predict.csv",
     n_trials: Annotated[
-        int, typer.Argument(help="Number of optimization trials.")
+        int, typer.Argument(help="Number of optimization trials per model.")
     ] = 100,
 ) -> None:
     _supported_type = (CSV, PARQUET)

diff --git a/rektgbm/encoder.py b/rektgbm/encoder.py
@@ -34,5 +34,5 @@ def fit_transform_label(self, series: YdataLike) -> YdataLike:
         self.fit_label(series=series)
         return self.transform_label(series=series)
 
-    def inverse_transform(self, series: XdataLike) -> XdataLike:
+    def inverse_transform(self, series: YdataLike) -> YdataLike:
         return self.label_encoder.inverse_transform(y=series)