From f9fd86ea8045267f3db9d9df2090f526161591fd Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Bojan=20Karla=C5=A1?= <bojan.karlas@gmail.com>
Date: Sat, 24 Feb 2024 14:30:45 -0500
Subject: [PATCH] Small update on the return type of the __call_ method of
 SklearnModelUtilityResult.

---
 datascope/importance/common.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/datascope/importance/common.py b/datascope/importance/common.py
index 1c5f38c..9581489 100644
--- 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()