typing bug

outbrain · Mar 19, 2024 · 54dd256 · 54dd256
1 parent 289f8eb
commit 54dd256
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/outrank/algorithms/importance_estimator.py b/outrank/algorithms/importance_estimator.py
@@ -223,7 +223,7 @@ def get_importances_estimate_nonmyopic(args: Any, tmp_df: pd.DataFrame):
     pass
 
 
-def initialize_classifier(surrogate_model: string):
+def initialize_classifier(surrogate_model: str):
     if 'surrogate-LR' in surrogate_model:
         return LogisticRegression(max_iter=100000)
     elif 'surrogate-SVM' in surrogate_model: