fix assignment warning

rdnfn · Nov 5, 2024 · d19ff5b · d19ff5b
1 parent d78d510
commit d19ff5b
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/inverse_cai/app/metrics.py b/src/inverse_cai/app/metrics.py
@@ -71,8 +71,10 @@ def compute_metrics(votes_df: pd.DataFrame, baseline_metrics: dict = None) -> di
     metrics = {}
 
     # slightly faster to make data types categorical
-    votes_df["principle"] = votes_df["principle"].astype("category")
-    votes_df["vote"] = votes_df["vote"].astype("category")
+    votes_df = votes_df.assign(
+        principle=votes_df["principle"].astype("category"),
+        vote=votes_df["vote"].astype("category"),
+    )
 
     # more efficient than doing operation for each principle group separately
     value_counts_all = (