Modify import in test_ks.py file

online-ml · Oct 30, 2023 · 994418f · 994418f
1 parent 8c65346
commit 994418f
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/river/stats/test_ks.py b/river/stats/test_ks.py
@@ -1,11 +1,12 @@
 from __future__ import annotations
 
 import numpy as np
-import river
 
-from collections import deque
 from scipy.stats import ks_2samp
 
+from collections import deque
+from river import stats
+
 
 def test_incremental_ks_statistics():
     initial_a = np.random.normal(loc=0, scale=1, size=5)
@@ -15,7 +16,7 @@ def test_incremental_ks_statistics():
     stream_b = np.random.normal(loc=1, scale=1, size=10)
 
     incremental_ks_statistics = []
-    incremental_ks = river.stats.KolmogorovSmirnov(statistic="ks")
+    incremental_ks = stats.KolmogorovSmirnov(statistic="ks")
     sliding_a = deque(initial_a)
     sliding_b = deque(initial_b)