Skip to content

Commit 42cd6d3

Browse files
committed
Fix minimum tests
1 parent 0c93c78 commit 42cd6d3

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

sdmetrics/single_table/privacy/disclosure_protection.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -86,9 +86,11 @@ def _get_null_categories(cls, real_data, synthetic_data, columns):
8686
@classmethod
8787
def _discretize_column(cls, real_column, synthetic_column, num_bins):
8888
bin_labels = [str(x) for x in range(num_bins)]
89-
real_binned, bins = pd.cut(real_column, num_bins, labels=bin_labels, retbins=True)
89+
real_binned, bins = pd.cut(
90+
real_column.to_numpy(), num_bins, labels=bin_labels, retbins=True
91+
)
9092
bins[0], bins[-1] = -np.inf, np.inf
91-
synthetic_binned = pd.cut(synthetic_column, bins, labels=bin_labels)
93+
synthetic_binned = pd.cut(synthetic_column.to_numpy(), bins, labels=bin_labels)
9294

9395
return real_binned.to_numpy(), synthetic_binned.to_numpy()
9496

0 commit comments

Comments
 (0)