UPDATE river version

online-ml · Jul 7, 2022 · 56645af · 56645af
1 parent c8bc4c8
commit 56645af
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/river_torch/anomaly/base.py b/river_torch/anomaly/base.py
@@ -4,13 +4,13 @@
 
 import pandas as pd
 import torch
-from river import anomaly, stats
+from river import stats, anomaly
 from torch import nn
 
 from river_torch.utils import dict2tensor, get_loss_fn, get_optimizer_fn
 
 
-class AutoEncoder(anomaly.AnomalyDetector, nn.Module):
+class AutoEncoder(anomaly.base.AnomalyDetector, nn.Module):
     """
     Base Auto Encoder
     ----------

diff --git a/setup.py b/setup.py
@@ -19,7 +19,7 @@
     "torch==1.11",
     "pandas~=1.3.2",
     "numpy>=1.21.6",
-    "river~=0.10.1",
+    "river~=0.11.1",
     "tqdm~=4.61.2",
     "pytest==7.0.1",
 ]