From e601c2dd116858e3173d845da4ae7157843364e4 Mon Sep 17 00:00:00 2001
From: Stefano Zamboni <39366866+SteZamboni@users.noreply.github.com>
Date: Fri, 19 Jul 2024 09:01:35 +0200
Subject: [PATCH] fix: added psi to enum drift dto (#124)

---
 api/app/models/metrics/drift_dto.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/api/app/models/metrics/drift_dto.py b/api/app/models/metrics/drift_dto.py
index 09ada9b2..cee2d971 100644
--- a/api/app/models/metrics/drift_dto.py
+++ b/api/app/models/metrics/drift_dto.py
@@ -10,6 +10,7 @@
 class DriftAlgorithm(str, Enum):
     KS = 'KS'
     CHI2 = 'CHI2'
+    PSI = 'PSI'
 
 
 class FeatureDriftCalculation(BaseModel):