fix: fixed alerts for retrocompatibility (#189)

* fix: fixed alerts for retrocompatibility * fix: fixed percentages for retrocompatibility * fix: update init_db.sql with update if conflict on UUID * fix: fixed init_db sql on init-data --------- Co-authored-by: dvalleri <[email protected]>
radicalbit · Oct 30, 2024 · d83d8d2 · d83d8d2
1 parent f36fa4a
commit d83d8d2
Show file tree

Hide file tree

Showing 2 changed files with 54 additions and 17 deletions.
diff --git a/api/app/services/model_service.py b/api/app/services/model_service.py
@@ -96,7 +96,7 @@ def get_summarized_percentages(self) -> TotPercentagesDTO:
         mq, mq_c = 0, 0
         dr, dr_c = 0, 0
         for _, metrics in models:
-            if metrics:
+            if metrics and metrics.percentages:
                 dq = dq + (
                     metrics.percentages['data_quality']['value']
                     if metrics.percentages['data_quality']['value'] >= 0
@@ -137,7 +137,7 @@ def get_last_n_alerts(self, n_alerts) -> List[AlertDTO]:
             latest_reference_dataset, latest_current_dataset = self.get_latest_datasets(
                 model.uuid
             )
-            if metrics:
+            if metrics and metrics.percentages:
                 for perc in ['data_quality', 'model_quality', 'drift']:
                     if count_alerts == n_alerts:
                         return res

diff --git a/init-data/init_db.sql b/init-data/init_db.sql