Skip to content

Commit 20cc6f7

Browse files
authored
Register summary stats in CLI runner (#353)
1 parent c6e5aa7 commit 20cc6f7

File tree

1 file changed

+10
-0
lines changed

1 file changed

+10
-0
lines changed

nannyml/runner.py

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -24,6 +24,11 @@
2424
from nannyml.performance_calculation import PerformanceCalculator
2525
from nannyml.performance_estimation.confidence_based import CBPE
2626
from nannyml.performance_estimation.direct_loss_estimation import DLE
27+
from nannyml.stats.avg.calculator import SummaryStatsAvgCalculator
28+
from nannyml.stats.count.calculator import SummaryStatsRowCountCalculator
29+
from nannyml.stats.median.calculator import SummaryStatsMedianCalculator
30+
from nannyml.stats.sum.calculator import SummaryStatsSumCalculator
31+
from nannyml.stats.std.calculator import SummaryStatsStdCalculator
2732

2833

2934
@dataclass
@@ -73,6 +78,11 @@ class CalculatorFactory:
7378
'dle': DLE,
7479
'missing_values': MissingValuesCalculator,
7580
'unseen_values': UnseenValuesCalculator,
81+
'summary_stats_avg': SummaryStatsAvgCalculator,
82+
'summary_stats_row_count': SummaryStatsRowCountCalculator,
83+
'summary_stats_median': SummaryStatsMedianCalculator,
84+
'summary_stats_std': SummaryStatsStdCalculator,
85+
'summary_stats_sum': SummaryStatsSumCalculator,
7686
}
7787

7888
@classmethod

0 commit comments

Comments
 (0)