fix(config): backwards compatibility

ing-bank · Jun 28, 2022 · 6891039 · 6891039
1 parent 5e92e32
commit 6891039
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/popmon/pipeline/metrics_pipelines.py b/popmon/pipeline/metrics_pipelines.py
@@ -240,7 +240,7 @@ def __init__(
                 suffix_mean="_mean",
                 suffix_std="_std",
                 suffix_pull="_pull",
-                metrics=[f"{reference_prefix}_{key}" for key in Comparisons.get_keys()],
+                metrics=[f"{reference_prefix}_{key}" for key in Comparisons.get_keys() if key in ['max_prob_diff', 'psi', 'jsd']],
             ),
             # 4. profiling of histograms, then pull calculation compared with reference mean and std,
             #        to obtain normalized residuals of profiles
@@ -303,7 +303,7 @@ def __init__(
                 suffix_mean="_mean",
                 suffix_std="_std",
                 suffix_pull="_pull",
-                metrics=[f"{reference_prefix}_{key}" for key in Comparisons.get_keys()],
+                metrics=[f"{reference_prefix}_{key}" for key in Comparisons.get_keys() if key in ['max_prob_diff', 'psi', 'jsd']],
             ),
             # 4. pull calculation compared with reference mean and std, to obtain normalized residuals of profiles
             ReferencePullCalculator(
@@ -357,7 +357,7 @@ def __init__(
                 suffix_mean="_mean",
                 suffix_std="_std",
                 suffix_pull="_pull",
-                metrics=[f"{reference_prefix}_{key}" for key in Comparisons.get_keys()],
+                metrics=[f"{reference_prefix}_{key}" for key in Comparisons.get_keys() if key in ['max_prob_diff', 'psi', 'jsd']],
             ),
             # 4. profiling of histograms, then pull calculation compared with reference mean and std,
             #        to obtain normalized residuals of profiles
@@ -415,7 +415,7 @@ def __init__(
                 suffix_mean="_mean",
                 suffix_std="_std",
                 suffix_pull="_pull",
-                metrics=[f"{reference_prefix}_{key}" for key in Comparisons.get_keys()],
+                metrics=[f"{reference_prefix}_{key}" for key in Comparisons.get_keys() if key in ['max_prob_diff', 'psi', 'jsd']],
             ),
             ExpandingPullCalculator(
                 read_key="profiles",