Skip to content

Commit b8859b1

Browse files
committed
lint + fix minimum version
1 parent 454d38e commit b8859b1

File tree

4 files changed

+9
-7
lines changed

4 files changed

+9
-7
lines changed

sdmetrics/multi_table/statistical/cardinality_shape_similarity.py

+3-1
Original file line numberDiff line numberDiff line change
@@ -3,10 +3,12 @@
33
import numpy as np
44
from scipy.stats import ks_2samp
55

6+
from sdmetrics._utils_metadata import (
7+
_validate_metadata_dict,
8+
)
69
from sdmetrics.goal import Goal
710
from sdmetrics.multi_table.base import MultiTableMetric
811
from sdmetrics.utils import get_cardinality_distribution
9-
from sdmetrics._utils_metadata import _validate_metadata, _validate_metadata_dict, _validate_single_table_metadata
1012

1113

1214
class CardinalityShapeSimilarity(MultiTableMetric):

sdmetrics/multi_table/statistical/cardinality_statistic_similarity.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -4,11 +4,11 @@
44

55
import numpy as np
66

7+
from sdmetrics._utils_metadata import _validate_metadata_dict
78
from sdmetrics.goal import Goal
89
from sdmetrics.multi_table.base import MultiTableMetric
910
from sdmetrics.utils import get_cardinality_distribution
1011
from sdmetrics.warnings import ConstantInputWarning
11-
from sdmetrics._utils_metadata import _validate_metadata_dict
1212

1313

1414
class CardinalityStatisticSimilarity(MultiTableMetric):

sdmetrics/timeseries/base.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -4,9 +4,9 @@
44

55
import pandas as pd
66

7+
from sdmetrics._utils_metadata import _validate_metadata_dict
78
from sdmetrics.base import BaseMetric
89
from sdmetrics.utils import get_columns_from_metadata
9-
from sdmetrics._utils_metadata import _validate_metadata_dict
1010

1111

1212
class TimeSeriesMetric(BaseMetric):

tests/integration/single_table/data_augmentation/test_binary_classifier_recall_efficacy.py

+4-4
Original file line numberDiff line numberDiff line change
@@ -124,7 +124,7 @@ def test_with_nan_target_column(self):
124124
)
125125

126126
# Assert
127-
assert result_breakdown['score'] in (0.6, 0.6538461538461539)
127+
assert result_breakdown['score'] in (0.3846153846153846, 0.6538461538461539)
128128

129129
def test_with_minority_being_majority(self):
130130
"""Test the metric when the minority class is the majority class."""
@@ -148,7 +148,7 @@ def test_with_minority_being_majority(self):
148148
)
149149

150150
# Assert
151-
assert score == 0.5
151+
assert score in (0.5, 0.3846153846153846)
152152

153153
def test_with_multi_class(self):
154154
"""Test the metric with multi-class classification.
@@ -175,7 +175,7 @@ def test_with_multi_class(self):
175175
)
176176

177177
# Assert
178-
assert score_breakdown['score'] in (0.46153846153846156, 0.5384615384615384)
178+
assert score_breakdown['score'] in (0.4230769230769231, 0.5384615384615384)
179179

180180
def test_speical_data_metadata_config(self):
181181
"""Test the metric with a special data and metadata configuration.
@@ -214,4 +214,4 @@ def test_speical_data_metadata_config(self):
214214
)
215215

216216
# Assert
217-
assert score == 0.5
217+
assert score in (0.5, 0.3846153846153846)

0 commit comments

Comments
 (0)