From 760236ab8bd6e861c29d5a69b7be5a7772e089ae Mon Sep 17 00:00:00 2001 From: Max Halford Date: Wed, 9 Aug 2023 23:21:44 +0200 Subject: [PATCH] Update knn_regressor.py --- river/neighbors/knn_regressor.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/river/neighbors/knn_regressor.py b/river/neighbors/knn_regressor.py index 0463047c29..0cc2bbf162 100644 --- a/river/neighbors/knn_regressor.py +++ b/river/neighbors/knn_regressor.py @@ -96,10 +96,8 @@ def _check_aggregation_method(self, method): """ if method not in {self._MEAN, self._MEDIAN, self._WEIGHTED_MEAN}: raise ValueError( - "Invalid aggregation_method: {}.\n" - "Valid options are: {}".format( - method, {self._MEAN, self._MEDIAN, self._WEIGHTED_MEAN} - ) + f"Invalid aggregation_method: {method}.\n" + f"Valid options are: {(self._MEAN, self._MEDIAN, self._WEIGHTED_MEAN)}" ) def learn_one(self, x, y):