Merge pull request #169 from IFCA/fix-mmd-streaming-chunk-size

Add missing chunk_size argument in MMD streaming

Author: Jaime Céspedes Sisniega

frouros/detectors/data_drift/batch/distance_based/mmd.py

@@ -51,9 +51,9 @@ def __init__(
     ) -> None:
         """Init method.
 
-        :param kernel: kernel function to use
+        :param kernel: kernel function
         :type kernel: Callable
-        :param chunk_size:
+        :param chunk_size: chunk size value
         :type chunk_size: Optional[int]
         :param callbacks: callbacks
         :type callbacks: Optional[Union[Callback, List[Callback]]]

frouros/detectors/data_drift/streaming/distance_based/mmd.py

@@ -28,13 +28,16 @@ class MMD(DistanceBasedBase):
     def __init__(
         self,
         kernel: Callable = rbf_kernel,
+        chunk_size: Optional[int] = None,
         callbacks: Optional[Union[Callback, List[Callback]]] = None,
         window_size: int = 10,
     ) -> None:
         """Init method.
 
         :param kernel: kernel function
         :type kernel: Callable
+        :param chunk_size: chunk size value
+        :type chunk_size: Optional[int]
         :param callbacks: callbacks
         :type callbacks: Optional[Union[Callback, List[Callback]]]
         :param window_size: window size
@@ -45,7 +48,10 @@ def __init__(
             statistical_type=MultivariateData(),
             callbacks=callbacks,
         )
-        self.mmd = MMDBatch(kernel=kernel)
+        self.mmd = MMDBatch(
+            kernel=kernel,
+            chunk_size=chunk_size,
+        )
         self.window_size = window_size
         self.X_queue = CircularQueue(max_len=self.window_size)