[skip ci] Support running domain classifier multivariate drift detect…

…ion from CLI and runner
NannyML · Jun 17, 2024 · 4f73494 · 4f73494
1 parent a24ab81
commit 4f73494
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/nannyml/runner.py b/nannyml/runner.py
@@ -19,6 +19,7 @@
 from nannyml.distribution.categorical import CategoricalDistributionCalculator
 from nannyml.distribution.continuous import ContinuousDistributionCalculator
 from nannyml.drift.multivariate.data_reconstruction import DataReconstructionDriftCalculator
+from nannyml.drift.multivariate.domain_classifier import DomainClassifierCalculator
 from nannyml.drift.univariate import UnivariateDriftCalculator
 from nannyml.exceptions import InvalidArgumentsException
 from nannyml.io import FileReader, FilesystemStore, Writer, WriterFactory
@@ -74,7 +75,8 @@ class CalculatorFactory:
 
     registry: Dict[str, Type] = {
         'univariate_drift': UnivariateDriftCalculator,
-        'multivariate_drift': DataReconstructionDriftCalculator,
+        'reconstruction_error': DataReconstructionDriftCalculator,
+        'domain_classifier': DomainClassifierCalculator,
         'performance': PerformanceCalculator,
         'cbpe': CBPE,
         'dle': DLE,