Merge pull request #267 from IFCA/fix-mmd-chunk-kwargs

Add missing MMD chunk_size in statistical kwargs
IFCA-Advanced-Computing · Aug 8, 2023 · 5f9c5ac · 5f9c5ac
2 parents 9f82542 + 5f21fe9
commit 5f9c5ac
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/frouros/detectors/data_drift/batch/distance_based/mmd.py b/frouros/detectors/data_drift/batch/distance_based/mmd.py
@@ -58,6 +58,7 @@ def __init__(  # noqa: D107
             statistical_method=self._mmd,
             statistical_kwargs={
                 "kernel": kernel,
+                "chunk_size": chunk_size,
             },
             callbacks=callbacks,
         )