Fixed type annotation

aai-institute · Nov 15, 2023 · 28ac351 · 28ac351
1 parent 1a2cb71
commit 28ac351
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/sensai/evaluation/evaluator.py b/src/sensai/evaluation/evaluator.py
@@ -415,7 +415,7 @@ class VectorClassificationModelEvaluator(VectorModelEvaluator[VectorClassificati
  def __init__(self,
  data: Optional[InputOutputData],
  test_data: InputOutputData = None,
- params: VectorClassificationModelEvaluatorParams = None):
+ params: ClassificationEvaluatorParams = None):
  """
  Constructs an evaluator with test and training data.