fix lint again

evidentlyai · Dec 17, 2024 · 98ab1f3 · 98ab1f3
1 parent 0579b2c
commit 98ab1f3
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/evidently/ui/dashboards/test_suites.py b/src/evidently/ui/dashboards/test_suites.py
@@ -161,10 +161,12 @@ def get_customdata(test: Test, date):
 
         def get_z_value(test, date) -> float:
             ti = points[date].get(test)
+            default = TEST_2_Z[TestStatus.SKIPPED]
+
             if ti is None:
-                return TEST_2_Z[TestStatus.SKIPPED]
+                return default
 
-            return TEST_2_Z.get(ti.status)
+            return TEST_2_Z.get(ti.status) or default
 
         fig = go.Figure(
             data=[