Skip to content

Commit f7aba3f

Browse files
committed
style: apply code formatting to drift.py
- Run black formatter to ensure consistent code style - Apply isort for proper import organization - Fix any linting issues with ruff - Maintain compatibility with existing scipy.stats imports
1 parent fa9d286 commit f7aba3f

File tree

1 file changed

+3
-4
lines changed

1 file changed

+3
-4
lines changed

‎giskard/testing/tests/drift.py‎

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@
99

1010
import numpy as np
1111
import pandas as pd
12-
from scipy.stats import chi2, ks_2samp, wasserstein_distance
12+
from scipy.stats import KstestResult, chi2, ks_2samp, wasserstein_distance
1313

1414
from giskard.core.test_result import TestMessage, TestMessageLevel, TestResult
1515
from giskard.datasets.base import Dataset
@@ -98,11 +98,10 @@ def _calculate_drift_psi(actual_series, reference_series, max_categories):
9898
return total_psi, pd.DataFrame(output_data)
9999

100100

101-
from scipy.stats import KstestResult
102-
103-
def _calculate_ks(actual_series, reference_series) -> "KstestResult":
101+
def _calculate_ks(actual_series, reference_series) -> KstestResult:
104102
return ks_2samp(reference_series, actual_series)
105103

104+
106105
def _calculate_earth_movers_distance(actual_series, reference_series):
107106
unique_reference = np.unique(reference_series)
108107
unique_actual = np.unique(actual_series)

0 commit comments

Comments
 (0)