added prefix removal to make sure parity metrics are interpreted corr…

…ectly
credo-ai · Oct 27, 2022 · 27e077c · 27e077c
1 parent e1d70e0
commit 27e077c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/credoai/lens/pipeline_creator.py b/credoai/lens/pipeline_creator.py
@@ -55,10 +55,10 @@ def process_evidence_requirements(evidence_requirements: List[EvidenceRequiremen
 
 
 def extract_metrics(labels):
-    """Extract metrics from a single evidence requirrment"""
+    """Extract metrics from a single evidence requirement"""
     metrics = set()
     if "metric_type" in labels:
-        metrics.add(labels["metric_type"])
+        metrics.add(labels["metric_type"].removesuffix('_parity'))
     elif "metric_types" in labels:
         metrics = metrics.union(labels["metric_types"])
     return metrics