Remove kwargs to be passed to callbacks. Set fixed variables instead

Author: jaime-cespedes-sisniega

frouros/callbacks/base.py

@@ -3,6 +3,8 @@
 import abc
 from typing import Optional
 
+import numpy as np  # type: ignore
+
 
 class BaseCallback(abc.ABC):
     """Abstract class representing a callback."""
@@ -55,11 +57,19 @@ def set_detector(self, detector) -> None:
     #         )
     #     self._detector = value
 
-    def on_fit_start(self, **kwargs) -> None:
-        """On fit start method."""
+    def on_fit_start(self, X: np.ndarray) -> None:  # noqa: N803
+        """On fit start method.
+
+        :param X: reference data
+        :type X: numpy.ndarray
+        """
 
-    def on_fit_end(self, **kwargs) -> None:
-        """On fit end method."""
+    def on_fit_end(self, X: np.ndarray) -> None:  # noqa: N803
+        """On fit end method.
+
+        :param X: reference data
+        :type X: numpy.ndarray
+        """
 
     @abc.abstractmethod
     def reset(self) -> None:

frouros/callbacks/batch/base.py

@@ -1,18 +1,44 @@
 """Base callback batch module."""
 
 import abc
+from typing import Any
+
+import numpy as np  # type: ignore
 
 from frouros.callbacks.base import BaseCallback
 
 
 class BaseCallbackBatch(BaseCallback):
     """Callback batch class."""
 
-    def on_compare_start(self, **kwargs) -> None:
-        """On compare start method."""
-
-    def on_compare_end(self, **kwargs) -> None:
-        """On compare end method."""
+    def on_compare_start(
+        self,
+        X_ref: np.ndarray,  # noqa: N803
+        X_test: np.ndarray,
+    ) -> None:
+        """On compare start method.
+
+        :param X_ref: reference data
+        :type X_ref: numpy.ndarray
+        :param X_test: test data
+        :type X_test: numpy.ndarray
+        """
+
+    def on_compare_end(
+        self,
+        result: Any,
+        X_ref: np.ndarray,  # noqa: N803
+        X_test: np.ndarray,
+    ) -> None:
+        """On compare end method.
+
+        :param result: result obtained from the `compare` method
+        :type result: Any
+        :param X_ref: reference data
+        :type X_ref: numpy.ndarray
+        :param X_test: test data
+        :type X_test: numpy.ndarray
+        """
 
     # FIXME: set_detector method as a workaround to  # pylint: disable=fixme
     #  avoid circular import problem. Make it an abstract method and

frouros/callbacks/streaming/base.py

@@ -1,18 +1,27 @@
 """Base callback streaming module."""
 
 import abc
+from typing import Union
 
 from frouros.callbacks.base import BaseCallback
 
 
 class BaseCallbackStreaming(BaseCallback):
     """Callback streaming class."""
 
-    def on_update_start(self, **kwargs) -> None:
-        """On update start method."""
+    def on_update_start(self, value: Union[int, float]) -> None:
+        """On update start method.
 
-    def on_update_end(self, **kwargs) -> None:
-        """On update end method."""
+        :param value: value used to update the detector
+        :type value: Union[int, float]
+        """
+
+    def on_update_end(self, value: Union[int, float]) -> None:
+        """On update end method.
+
+        :param value: value used to update the detector
+        :type value: Union[int, float]
+        """
 
     # FIXME: set_detector method as a workaround to  # pylint: disable=fixme
     #  avoid circular import problem. Make it an abstract method and

frouros/detectors/concept_drift/base.py

@@ -184,17 +184,17 @@ def update(self, value: Union[int, float], **kwargs) -> Dict[str, Any]:
 
         :param value: value to update detector
         :type value: Union[int, float]
+        :return: callbacks logs
+        :rtype: Dict[str, Any]]
         """
         for callback in self.callbacks:  # type: ignore
             callback.on_update_start(  # type: ignore
                 value=value,
-                **kwargs,
             )
         self._update(value=value, **kwargs)
         for callback in self.callbacks:  # type: ignore
             callback.on_update_end(  # type: ignore
                 value=value,
-                **kwargs,
             )
 
         callbacks_logs = self._get_callbacks_logs()

frouros/detectors/data_drift/base.py

@@ -187,17 +187,15 @@ def fit(self, X: np.ndarray, **kwargs) -> Dict[str, Any]:  # noqa: N803
         for callback in self.callbacks:  # type: ignore
             callback.on_fit_start(
                 X=X,
-                **kwargs,
             )
         self._fit(X=X, **kwargs)
         for callback in self.callbacks:  # type: ignore
             callback.on_fit_end(
                 X=X,
-                **kwargs,
             )
 
-        logs = self._get_callbacks_logs()
-        return logs
+        callbacks_logs = self._get_callbacks_logs()
+        return callbacks_logs
 
     def reset(self) -> None:
         """Reset method."""

frouros/detectors/data_drift/batch/base.py

@@ -60,7 +60,7 @@ def compare(
     ) -> Tuple[np.ndarray, Dict[str, Any]]:
         """Compare values.
 
-        :param X: feature data
+        :param X: test data
         :type X: numpy.ndarray
         :return: compare result and callbacks logs
         :rtype: Tuple[numpy.ndarray, Dict[str, Any]]

frouros/detectors/data_drift/streaming/base.py

@@ -56,27 +56,29 @@ def reset(self) -> None:
         self._reset()
 
     def update(
-        self, value: Union[int, float]
+        self,
+        value: Union[int, float],
     ) -> Tuple[Optional[BaseResult], Dict[str, Any]]:
         """Update detector.
 
         :param value: value to use to update the detector
         :type value: Union[int, float]
-        :return: update result
-        :rtype: Optional[BaseResult]
+        :return: update result and callbacks logs
+        :rtype: Tuple[Optional[BaseResult], Dict[str, Any]]
         """
         self._common_checks()  # noqa: N806
         self._specific_checks(X=value)  # noqa: N806
         self.num_instances += 1
 
         for callback in self.callbacks:  # type: ignore
-            callback.on_update_start(value=value)  # type: ignore
+            callback.on_update_start(  # type: ignore
+                value=value,  # type: ignore
+            )
         result = self._update(value=value)
-        if result is not None:
-            for callback in self.callbacks:  # type: ignore
-                callback.on_update_end(  # type: ignore
-                    value=result.distance,  # type: ignore
-                )
+        for callback in self.callbacks:  # type: ignore
+            callback.on_update_end(  # type: ignore
+                value=result,  # type: ignore
+            )
 
         callbacks_logs = self._get_callbacks_logs()
         return result, callbacks_logs