diff --git a/deep_river/base.py b/deep_river/base.py
index 96db5d8..c69a6fb 100644
--- a/deep_river/base.py
+++ b/deep_river/base.py
@@ -72,7 +72,7 @@ def __init__(
 
     @abc.abstractmethod
     def learn_one(
-        self, x: dict, y: Optional[Union[Any, Dict[Any, Any]]]
+        self, x: dict, y: Optional[Union[Any, Dict[Any, Any]]], **kwargs
     ) -> "DeepEstimator":
         """
         Performs one step of training with a single example.