validmind 1.22.6__tar.gz → 1.22.7__tar.gz

{validmind-1.22.6 → validmind-1.22.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: validmind
-Version: 1.22.6
+Version: 1.22.7
 Summary: ValidMind Developer Framework
 License: Commercial License
 Author: Andres Rodriguez

{validmind-1.22.6 → validmind-1.22.7}/pyproject.toml

@@ -10,7 +10,7 @@ description = "ValidMind Developer Framework"
 license = "Commercial License"
 name = "validmind"
 readme = "README.pypi.md"
-version = "1.22.6"
+version = "1.22.7"
 
 [tool.poetry.dependencies]
 aiohttp = {extras = ["speedups"], version = "^3.8.4"}

validmind-1.22.7/validmind/__version__.py

@@ -0,0 +1 @@
+__version__ = "1.22.7"

{validmind-1.22.6 → validmind-1.22.7}/validmind/tests/model_validation/sklearn/PopulationStabilityIndex.py

@@ -86,7 +86,9 @@ class PopulationStabilityIndex(Metric):
         # The data looks like this: [{"initial": 2652, "percent_initial": 0.5525, "new": 830, "percent_new": 0.5188, "psi": 0.0021},...
         psi_table = [
             {
-                "Bin": i,
+                "Bin": i
+                if i < (len(metric_value) - 1)
+                else "Total",  # The last bin is the "Total" bin
                 "Count Initial": values["initial"],
                 "Percent Initial (%)": values["percent_initial"] * 100,
                 "Count New": values["new"],
@@ -230,4 +232,14 @@ class PopulationStabilityIndex(Metric):
             figure=fig,
         )
 
+        # Calculate the sum of each numeric column
+        total_psi = {
+            key: sum(d.get(key, 0) for d in psi_results)
+            for key in psi_results[0].keys()
+            if isinstance(psi_results[0][key], (int, float))
+        }
+
+        # Add the total PSI dictionary to the list
+        psi_results.append(total_psi)
+
         return self.cache_results(metric_value=psi_results, figures=[figure])

validmind-1.22.6/validmind/__version__.py

@@ -1 +0,0 @@
-__version__ = "1.22.6"