ruff

online-ml · Dec 2, 2023 · 715942d · 715942d
1 parent 4dff1df
commit 715942d
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 3 deletions.
diff --git a/river/metrics/base.py b/river/metrics/base.py
@@ -77,7 +77,7 @@ class ClassificationMetric(Metric):
 
     _fmt = ".2%"  # output a percentage, e.g. 0.427 becomes "42,7%"
 
-    def __init__(self, cm: "confusion.ConfusionMatrix" | None = None):
+    def __init__(self, cm=None):
         # HACK: there is a circular dependency between ConfusionMatrix and ClassificationMetric. We
         # use ConfusionMatrix here so as to express metrics in terms of false/true
         # positives/negatives. But for UX reasons, we also want ConfusionMatrix to be a

diff --git a/river/metrics/confusion.py b/river/metrics/confusion.py
@@ -3,8 +3,7 @@
 import functools
 from collections import defaultdict
 
-from river import metrics
-from river import utils
+from river import metrics, utils
 
 
 class ConfusionMatrix(metrics.base.MultiClassMetric):

diff --git a/river/stats/_rust_stats.pyi b/river/stats/_rust_stats.pyi
@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 class RsQuantile:
     def __init__(self, q: float): ...
     def update(self, x: float): ...