diff --git a/nannyml/runner.py b/nannyml/runner.py index b1280602..d2eedac4 100644 --- a/nannyml/runner.py +++ b/nannyml/runner.py @@ -26,6 +26,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 @@ -77,6 +82,11 @@ class CalculatorFactory: 'unseen_values': UnseenValuesCalculator, 'continuous_distribution': ContinuousDistributionCalculator, 'categorical_distribution': CategoricalDistributionCalculator, + 'summary_stats_avg': SummaryStatsAvgCalculator, + 'summary_stats_row_count': SummaryStatsRowCountCalculator, + 'summary_stats_median': SummaryStatsMedianCalculator, + 'summary_stats_std': SummaryStatsStdCalculator, + 'summary_stats_sum': SummaryStatsSumCalculator, } @classmethod