Automatic pre-commit fixes

Author: PravasMohanty

aeon/visualisation/results/_mcm.py

@@ -5,9 +5,10 @@
 __all__ = ["create_multi_comparison_matrix"]
 
 import json
+import logging
 import os
 from typing import Dict, List, Optional, Union
-import logging
+
 import matplotlib.pyplot as plt
 import numpy as np
 import pandas as pd
@@ -19,28 +20,29 @@
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
+
 def create_multi_comparison_matrix(
-   df_results: Union[str, pd.DataFrame],
+    df_results: Union[str, pd.DataFrame],
     output_dir: str = "./",
-    save_files: Optional[Dict[str, str]] = None,
+    save_files: Optional[dict[str, str]] = None,
     used_statistic: str = "Accuracy",
     save_as_json: bool = True,
     plot_1v1_comparisons: bool = False,
-    order_settings: Dict[str, str] = {"win_tie_loss": "higher", "better": "decreasing"},
+    order_settings: dict[str, str] = {"win_tie_loss": "higher", "better": "decreasing"},
     include_pvalue: bool = True,
     pvalue_test: str = "wilcoxon",
-    pvalue_test_params: Optional[Dict[str, str]] = None,
+    pvalue_test_params: Optional[dict[str, str]] = None,
     pvalue_correction: Optional[str] = None,
     pvalue_threshold: float = 0.05,
     use_mean: str = "mean-difference",
     order_stats: str = "average-statistic",
     dataset_column: Optional[str] = None,
     precision: int = 4,
     load_analysis: bool = False,
-    row_comparates: Optional[List[str]] = None,
-    col_comparates: Optional[List[str]] = None,
-    excluded_row_comparates: Optional[List[str]] = None,
-    excluded_col_comparates: Optional[List[str]] = None,
+    row_comparates: Optional[list[str]] = None,
+    col_comparates: Optional[list[str]] = None,
+    excluded_row_comparates: Optional[list[str]] = None,
+    excluded_col_comparates: Optional[list[str]] = None,
     colormap: str = "coolwarm",
     fig_size: Union[str, tuple] = "auto",
     font_size: Union[str, int] = "auto",
@@ -65,9 +67,9 @@ def create_multi_comparison_matrix(
         row should contain the names of the estimators and the first column can
         contain the names of the problems if `dataset_column` is true.
     output_dir: str, default = './'
-        The output directory for the results.        
+        The output directory for the results.
     save_files: dict, default = None
-        Dictionary to handle all save options (e.g., {"pdf": "output.pdf", "png": "output.png"}).        
+        Dictionary to handle all save options (e.g., {"pdf": "output.pdf", "png": "output.png"}).
     used_statistic: str, default = 'Score'
         Name of the metric being assesses (e.g. accuracy, error, mse).
     save_as_json: bool, default = True
@@ -167,9 +169,8 @@ def create_multi_comparison_matrix(
     Evaluations That Is Stable Under Manipulation Of The Comparate Set
     arXiv preprint arXiv:2305.11921, 2023.
     """
-    
     logger.info("Starting MCM creation...")
-    
+
     if isinstance(df_results, str):
         try:
             df_results = pd.read_csv(df_results)
@@ -179,7 +180,7 @@ def create_multi_comparison_matrix(
             raise ValueError(f"The file {df_results} is empty.")
         except Exception as e:
             raise ValueError(f"An error occurred while reading the file: {e}")
-    
+
     analysis = _get_analysis(
         df_results,
         output_dir=output_dir,
@@ -203,7 +204,7 @@ def create_multi_comparison_matrix(
         excluded_row_comparates=excluded_row_comparates,
         excluded_col_comparates=excluded_col_comparates,
     )
-    
+
     # Generate figure
     fig = _draw(
         analysis,
@@ -218,7 +219,6 @@ def create_multi_comparison_matrix(
         include_legend=include_legend,
         show_symmetry=show_symmetry,
     )
-    
 
     # start drawing heatmap
     if save_files:
@@ -239,7 +239,6 @@ def create_multi_comparison_matrix(
     return fig
 
 
-
 def _get_analysis(
     df_results,
     output_dir="./",