From 28ac351e28ed554554720aae7b8bf596b98938d0 Mon Sep 17 00:00:00 2001 From: Dominik Jain Date: Wed, 15 Nov 2023 17:54:17 +0100 Subject: [PATCH] Fixed type annotation --- src/sensai/evaluation/evaluator.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/sensai/evaluation/evaluator.py b/src/sensai/evaluation/evaluator.py index f2c9f9b78..a648d5cfd 100644 --- 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.