Fix Compatibility issue (#165)

* Add BaseEstimator inheritance

* Format files with new black version

* chore: drop Python 3.9 support (require >=3.10) and update CI matrix

---------

Co-authored-by: Huy Pham <hpham@talosix.com>
Co-authored-by: Fábio Malcher Miranda <fabio.miranda@ifap.edu.br>

Author: 3 people

.github/workflows/test-pr.yml

@@ -15,7 +15,7 @@ jobs:
     strategy:
       fail-fast: false
       matrix:
-        python-version: ["3.9", "3.10", "3.11", "3.12"]
+        python-version: ["3.10", "3.11", "3.12"]
         os: [ubuntu-latest, macOS-latest, windows-latest]
     steps:
     - uses: actions/checkout@v2

.pre-commit-config.yaml

@@ -6,7 +6,7 @@ repos:
       - id: end-of-file-fixer
       - id: trailing-whitespace
   - repo: https://github.com/psf/black
-    rev: 24.2.0
+    rev: 26.3.0
     hooks:
       - id: black
   - repo: https://github.com/pycqa/flake8

benchmarks/consumer_complaints/scripts/data.py

@@ -1,5 +1,6 @@
 #!/usr/bin/env python3
 """Script to share common functions for data manipulation."""
+
 import csv
 from typing import TextIO
 

benchmarks/consumer_complaints/scripts/metrics.py

@@ -1,5 +1,6 @@
 #!/usr/bin/env python3
 """Script to compute metrics."""
+
 import argparse
 import sys
 from argparse import Namespace

benchmarks/consumer_complaints/scripts/predict.py

@@ -1,5 +1,6 @@
 #!/usr/bin/env python3
 """Script to predict with flat approach."""
+
 import argparse
 import pickle
 import sys

benchmarks/consumer_complaints/scripts/split_data.py

@@ -1,5 +1,6 @@
 #!/usr/bin/env python3
 """Script to split train and test data."""
+
 import argparse
 import sys
 from argparse import Namespace

benchmarks/consumer_complaints/scripts/statistics.py

@@ -1,5 +1,6 @@
 #!/usr/bin/env python3
 """Script to store basic statistical information."""
+
 import argparse
 import os
 import sys

benchmarks/consumer_complaints/scripts/train.py

@@ -1,5 +1,6 @@
 #!/usr/bin/env python3
 """Script to train with flat or hierarchical approaches."""
+
 import argparse
 import pickle
 import sys

benchmarks/consumer_complaints/scripts/tune.py

@@ -1,5 +1,6 @@
 #!/usr/bin/env python3
 """Script to perform hyper-parameter tuning for flat and hierarchical approaches."""
+
 import hashlib
 import json
 import logging
@@ -170,7 +171,7 @@ def load_trial(cfg: DictConfig) -> List[float]:
     md5 = compute_md5(hyperparameters)
     filename = f"{cfg.output_dir}/{md5}.sav"
     if os.path.exists(filename):
-        (_, scores) = pickle.load(open(filename, "rb"))
+        _, scores = pickle.load(open(filename, "rb"))
         log.info(f"Loaded trial with F-scores {scores}")
         return scores
     else:

benchmarks/consumer_complaints/scripts/tune_table.py

@@ -1,5 +1,6 @@
 #!/usr/bin/env python3
 """Script to create table with tuning results for flat and hierarchical approaches."""
+
 import argparse
 import glob
 import pickle