Register summary stats in CLI runner

NannyML · Jan 17, 2024 · ae71f76 · ae71f76
1 parent aad7edf
commit ae71f76
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/nannyml/runner.py b/nannyml/runner.py
@@ -24,6 +24,11 @@
 from nannyml.performance_calculation import PerformanceCalculator
 from nannyml.performance_estimation.confidence_based import CBPE
 from nannyml.performance_estimation.direct_loss_estimation import DLE
+from nannyml.stats.avg.calculator import SummaryStatsAvgCalculator
+from nannyml.stats.count.calculator import SummaryStatsRowCountCalculator
+from nannyml.stats.median.calculator import SummaryStatsMedianCalculator
+from nannyml.stats.sum.calculator import SummaryStatsSumCalculator
+from nannyml.stats.std.calculator import SummaryStatsStdCalculator
 
 
 @dataclass
@@ -73,6 +78,11 @@ class CalculatorFactory:
         'dle': DLE,
         'missing_values': MissingValuesCalculator,
         'unseen_values': UnseenValuesCalculator,
+        'summary_stats_avg': SummaryStatsAvgCalculator,
+        'summary_stats_row_count': SummaryStatsRowCountCalculator,
+        'summary_stats_median': SummaryStatsMedianCalculator,
+        'summary_stats_std': SummaryStatsStdCalculator,
+        'summary_stats_sum': SummaryStatsSumCalculator,
     }
 
     @classmethod