resolve some warnings

glassonion1 · Sep 20, 2024 · 28ea9aa · 28ea9aa
1 parent 3e61127
commit 28ea9aa
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/anonypy/anonymity.py b/anonypy/anonymity.py
@@ -13,7 +13,9 @@ def is_t_close(df, partition, sensitive_column, global_freqs, p):
     total_count = float(len(partition))
     d_max = None
     group_counts = (
-        df.loc[partition].groupby(sensitive_column)[sensitive_column].agg("count")
+        df.loc[partition]
+        .groupby(sensitive_column, observed=True)[sensitive_column]
+        .agg("count")
     )
     for value, count in group_counts.to_dict().items():
         p = count / total_count
@@ -26,7 +28,9 @@ def is_t_close(df, partition, sensitive_column, global_freqs, p):
 def get_global_freq(df, sensitive_column):
     global_freqs = {}
     total_count = float(len(df))
-    group_counts = df.groupby(sensitive_column)[sensitive_column].agg("count")
+    group_counts = df.groupby(sensitive_column, observed=True)[sensitive_column].agg(
+        "count"
+    )
 
     for value, count in group_counts.to_dict().items():
         p = count / total_count