fix tests

Author: Samoed

autointent/_dump_tools.py

@@ -254,8 +254,8 @@ def load(  # noqa: C901, PLR0912, PLR0915
                         msg = f"Error loading HF tokenizer {tokenizer_dir.name}: {e}"
                         logger.exception(msg)
             elif child.name == Dumper.catboost_models:
-                cat_model = CatBoostClassifier(str(path / Dumper.catboost_models / "model.cbm"))
-                cat_model.load_model()
+                cat_model = CatBoostClassifier()
+                cat_model.load_model(str(path / Dumper.catboost_models / "model.cbm"))
             else:
                 msg = f"Found unexpected child {child}"
                 logger.error(msg)

autointent/modules/scoring/_catboost/catboost_scorer.py

@@ -1,4 +1,5 @@
 """CatBoostScorer class for CatBoost-based classification with switchable encoding."""
+
 import logging
 from enum import StrEnum
 from typing import Any, cast
@@ -60,7 +61,7 @@ class CatBoostScorer(BaseScorer):
         eval_metric="Accuracy",
         random_seed=42,
         verbose=False,
-        features_type="text",  # or "embedding" or "both"
+        features_type="embedding",  # or "text" or "both"
     )
     utterances = ["hello", "goodbye", "allo", "sayonara"]
     labels = [0, 1, 0, 1]
@@ -71,8 +72,8 @@ class CatBoostScorer(BaseScorer):
 
     .. testoutput::
 
-        [[0.50525691 0.49474309]
-         [0.50525691 0.49474309]]
+        [[0.41493207 0.58506793]
+         [0.55036046 0.44963954]]
 
     """
 

tests/modules/scoring/test_catboost.py

@@ -78,39 +78,14 @@ def test_catboost_prediction_multilabel(dataset):
         predictions,
         np.array(
             [
-                [
-                    0.2,
-                    0.7,
-                    0.2,
-                    0.2,
-                ],
-                [
-                    0.2,
-                    0.4,
-                    0.2,
-                    0.4,
-                ],
-                [
-                    0.2,
-                    0.6,
-                    0.2,
-                    0.2,
-                ],
-                [
-                    0.2,
-                    0.3,
-                    0.2,
-                    0.4,
-                ],
-                [
-                    0.1,
-                    0.8,
-                    0.1,
-                    0.1,
-                ],
+                [0.41777172, 0.5278134, 0.41807876, 0.4174544],
+                [0.40775846, 0.46434019, 0.42728555, 0.43836945],
+                [0.4207232, 0.49201536, 0.42798494, 0.41541217],
+                [0.46765036, 0.45065999, 0.49705517, 0.45052473],
+                [0.41694272, 0.54160408, 0.40944069, 0.41674984],
             ]
         ),
-        0.1,
+        rtol=0.01,
     )