fix(plots): correct type of slot & account for None mainstat

trumully · Apr 17, 2024 · 0948acb · 0948acb
1 parent 1918b1a
commit 0948acb
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/artipy/analysis/plots.py b/src/artipy/analysis/plots.py
@@ -168,8 +168,8 @@ def plot_expected_against_actual_mainstats(iterations: int = 1000) -> None:
     }
 
     for a in artifacts:
-        slot: ArtifactSlot = ArtifactSlot(a.artifact_slot)
-        if slot in expected_mainstats:
+        slot: ArtifactSlot = ArtifactSlot(str(a.artifact_slot))
+        if slot in expected_mainstats and a.mainstat is not None:
             actual_mainstats[slot].append(a.mainstat.name)
 
     actual_mainstats_pct: dict[ArtifactSlot, dict[StatType, float]] = {