Browse Source

solve backwards compability

ogert 2 years ago
parent
commit
301aab81fb
1 changed files with 5 additions and 3 deletions
  1. 5 3
      cdplib/ml_validation/fine_tuning.py

+ 5 - 3
cdplib/ml_validation/fine_tuning.py

@@ -10,7 +10,7 @@ import sys
 
 import numpy as np
 
-from numpy.typing import ArrayLike
+# from numpy.typing import ArrayLike
 
 if (sys.version_info.major == 3) & (sys.version_info.minor >= 8):
     from typing import Iterable, Callable
@@ -19,8 +19,10 @@ else:
 
 
 def get_optimal_proba_threshold(score_func: Callable,
-                                y_true: ArrayLike,
-                                proba: ArrayLike,
+                                # y_true: ArrayLike,
+                                # proba: ArrayLike,
+                                y_true,
+                                proba,
                                 threshold_set: Iterable = None):
     """
     """