Small update on the return type of the __call_ method of SklearnModel…

…UtilityResult.
easeml · Feb 24, 2024 · f9fd86e · f9fd86e
1 parent 66192ea
commit f9fd86e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/datascope/importance/common.py b/datascope/importance/common.py
@@ -337,7 +337,7 @@ def __call__(
         metadata_test: Optional[Union[NDArray, DataFrame]] = None,
         null_score: Optional[float] = None,
         seed: int = DEFAULT_SEED,
-    ) -> UtilityResult:
+    ) -> SklearnModelUtilityResult:
         result = SklearnModelUtilityResult()
         if not isinstance(y_test, ndarray):
             y_test = y_test.to_numpy()