fix: added psi to enum drift dto (#124)

radicalbit · Jul 19, 2024 · e601c2d · e601c2d
1 parent 67e94e9
commit e601c2d
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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):