Merge branch 'online-ml:master' into predict_in_inference_mode

online-ml · Dec 20, 2022 · 11d3a6d · 11d3a6d
2 parents 8049ecb + aec0316
commit 11d3a6d
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/river_torch/anomaly/rolling_ae.py b/river_torch/anomaly/rolling_ae.py
@@ -154,7 +154,9 @@ def _learn(self, x: torch.Tensor):
         loss.backward()
         self.optimizer.step()
 
-    def learn_one(self, x: dict, y: None) -> "RollingAutoencoder":
+    def learn_one(
+        self, x: dict, y: Any = None, **kwargs
+    ) -> "RollingAutoencoder":
         """
         Performs one step of training with a single example.