refactorings v2

outbrain-inc · Jan 5, 2024 · 1a1d7b6 · 1a1d7b6
1 parent d5fc1c2
commit 1a1d7b6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/outrank/core_ranking.py b/outrank/core_ranking.py
@@ -415,7 +415,7 @@ def compute_value_counts(input_dataframe: pd.DataFrame, args: Any):
         del GLOBAL_RARE_VALUE_STORAGE[to_remove_val]
 
 
-def compute_cardinalities(input_dataframe: pd.DataFrame, pbar: Any, args: Any) -> None:
+def compute_cardinalities(input_dataframe: pd.DataFrame, pbar: Any) -> None:
     """Compute cardinalities of features, incrementally"""
 
     global GLOBAL_CARDINALITY_STORAGE
@@ -542,7 +542,7 @@ def compute_batch_ranking(
     feature_memory_consumption = compute_feature_memory_consumption(
         input_dataframe, args,
     )
-    compute_cardinalities(input_dataframe, pbar, args)
+    compute_cardinalities(input_dataframe, pbar)
 
     if args.task == 'identify_rare_values':
         compute_value_counts(input_dataframe, args)