Minor Type Annotation Fix

Signed-off-by: Nikolas Theologitis <[email protected]>
Trusted-AI · Oct 17, 2024 · 3f4c854 · 3f4c854
1 parent 4026523
commit 3f4c854
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/aix360/algorithms/glance/iterative_merges/iterative_merges.py b/aix360/algorithms/glance/iterative_merges/iterative_merges.py
@@ -178,7 +178,7 @@ def fit(
         min_cost_eff_thres__effectiveness_threshold: Optional[float] = None,
         min_cost_eff_thres_combinations__num_min_cost: Optional[int] = None,
         eff_thres_hybrid__max_n_actions_full_combinations: Optional[int] = None,
-    ) -> "IterativeMerges":
+    ) -> "C_GLANCE":
         """
         Fits the clustering and counterfactual generation model to the provided dataset.