Update knn_regressor.py

online-ml · Aug 9, 2023 · 760236a · 760236a
1 parent fb8d71e
commit 760236a
Showing 1 changed file with 2 additions and 4 deletions.
diff --git 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):