bug fix

joshuawe · Nov 10, 2023 · 9e91555 · 9e91555
1 parent c5a72df
commit 9e91555
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/plotsandgraphs/binary_classifier.py b/plotsandgraphs/binary_classifier.py
@@ -368,7 +368,7 @@ def plot_roc_curve(
 
 def plot_calibration_curve(
     y_prob: np.ndarray, y_true: np.ndarray, n_bins=10, save_fig_path=None
-):
+) -> Figure:
     """
     Creates calibration plot for a binary classifier and calculates the ECE.
 
@@ -462,7 +462,7 @@ def plot_calibration_curve(
         path.parent.mkdir(parents=True, exist_ok=True)
         fig.savefig(save_fig_path, bbox_inches="tight")
 
-    return fig, ece
+    return fig
 
 
 def plot_y_prob_histogram(y_prob: np.ndarray, y_true: Optional[np.ndarray]=None, save_fig_path=None) -> Figure: