Pre-commit autoupdate (#759)

* Pre-commit autoupdate

* Hooks

---------

Co-authored-by: quant-ranger[bot] <132915763+quant-ranger[bot]@users.noreply.github.com>
Co-authored-by: lbittarello <luca.bittarello@gmail.com>
Co-authored-by: Marc-Antoine Schmidt <marc-antoine.schmidt@quantco.com>

.pre-commit-config.yaml

@@ -1,13 +1,13 @@
 repos:
  - repo: https://github.com/Quantco/pre-commit-mirrors-black
-   rev: 23.12.1
+   rev: 24.1.1
    hooks:
      - id: black-conda
        args:
          - --safe
          - --target-version=py39
  - repo: https://github.com/Quantco/pre-commit-mirrors-flake8
-   rev: 6.1.0
+   rev: 7.0.0
    hooks:
     - id: flake8-conda
       additional_dependencies: [

setup.py

@@ -74,9 +74,9 @@
     package_dir={"": "src"},
     packages=find_packages(
         where="src",
-        include=["glum"]
-        if os.environ.get("CONDA_BUILD")
-        else ["glum", "glum_benchmarks"],
+        include=(
+            ["glum"] if os.environ.get("CONDA_BUILD") else ["glum", "glum_benchmarks"]
+        ),
     ),
     python_requires=">=3.9",
     install_requires=[
@@ -88,13 +88,15 @@
         "scipy",
         "tabmat>=3.1.0,<4.0.0",
     ],
-    entry_points=None
-    if os.environ.get("CONDA_BUILD")
-    else """
+    entry_points=(
+        None
+        if os.environ.get("CONDA_BUILD")
+        else """
         [console_scripts]
         glm_benchmarks_run = glum_benchmarks.cli_run:cli_run
         glm_benchmarks_analyze = glum_benchmarks.cli_analyze:cli_analyze
-    """,
+    """
+    ),
     ext_modules=cythonize(
         ext_modules,
         annotate=False,

src/glum/_distribution.py

@@ -956,9 +956,7 @@ def unit_deviance(self, y: np.ndarray, mu: np.ndarray) -> np.ndarray:
         -------
         array-like
         """
-        return 2 * y * (np.arctan(y) - np.arctan(mu)) + np.log(
-            (1 + mu**2) / (1 + y**2)
-        )
+        return 2 * y * (np.arctan(y) - np.arctan(mu)) + np.log((1 + mu**2) / (1 + y**2))
 
 
 class BinomialDistribution(ExponentialDispersionModel):

src/glum/_glm.py

@@ -2258,11 +2258,15 @@ def _set_up_and_check_fit_args(
             if any(X.dtypes == "category"):
                 self.feature_names_ = list(
                     chain.from_iterable(
-                        _name_categorical_variables(
-                            dtype.categories, column, getattr(self, "drop_first", False)
+                        (
+                            _name_categorical_variables(
+                                dtype.categories,
+                                column,
+                                getattr(self, "drop_first", False),
+                            )
+                            if isinstance(dtype, pd.CategoricalDtype)
+                            else [column]
                         )
-                        if isinstance(dtype, pd.CategoricalDtype)
-                        else [column]
                         for column, dtype in zip(X.columns, X.dtypes)
                     )
                 )
@@ -2289,9 +2293,14 @@ def _expand_categorical_penalties(penalty, X, drop_first):
                         return np.array(
                             list(
                                 chain.from_iterable(
-                                    [elmt for _ in dtype.categories[int(drop_first) :]]
-                                    if isinstance(dtype, pd.CategoricalDtype)
-                                    else [elmt]
+                                    (
+                                        [
+                                            elmt
+                                            for _ in dtype.categories[int(drop_first) :]
+                                        ]
+                                        if isinstance(dtype, pd.CategoricalDtype)
+                                        else [elmt]
+                                    )
                                     for elmt, dtype in zip(penalty, X.dtypes)
                                 )
                             )

src/glum_benchmarks/bench_liblinear.py

@@ -77,9 +77,11 @@ def liblinear_bench(
     model_args = dict(
         penalty=pen,
         tol=benchmark_convergence_tolerance,
-        C=1 / (X.shape[0] * alpha)
-        if reg_multiplier is None
-        else 1 / (X.shape[0] * alpha * reg_multiplier),
+        C=(
+            1 / (X.shape[0] * alpha)
+            if reg_multiplier is None
+            else 1 / (X.shape[0] * alpha * reg_multiplier)
+        ),
         # Note that when an intercept is fitted, it is subject to regularization, unlike
         # other solvers. intercept_scaling helps combat this by inflating the intercept
         # column, though too low of a value leaves too much regularization and too high

src/glum_benchmarks/cli_run.py

@@ -81,12 +81,14 @@ def cli_run(
         for Ln in libraries.keys():
             click.echo(f"running problem={Pn} library={Ln}")
             new_params = params.update_params(problem_name=Pn, library_name=Ln)
-            result, regularization_strength_ = execute_problem_library(
+            result, _ = execute_problem_library(
                 new_params,
                 iterations,
-                defaults["diagnostics_level"]
-                if params.diagnostics_level is None
-                else params.diagnostics_level,  # type: ignore
+                (
+                    defaults["diagnostics_level"]  # type: ignore
+                    if params.diagnostics_level is None  # type: ignore
+                    else params.diagnostics_level  # type: ignore
+                ),
             )
             _save_benchmark_results(
                 output_dir,

src/glum_benchmarks/orig_sklearn_fork/_glm.py

@@ -889,9 +889,7 @@ def unit_variance_derivative(self, mu):
         return 2 * mu
 
     def unit_deviance(self, y, mu):
-        return 2 * y * (np.arctan(y) - np.arctan(mu)) + np.log(
-            (1 + mu**2) / (1 + y**2)
-        )
+        return 2 * y * (np.arctan(y) - np.arctan(mu)) + np.log((1 + mu**2) / (1 + y**2))
 
 
 class BinomialDistribution(ExponentialDispersionModel):

src/glum_benchmarks/util.py

@@ -409,9 +409,7 @@ def clear_cache(force=False):
     if cache_location is None:
         return
 
-    cache_size_limit = float(
-        os.environ.get("GLM_BENCHMARKS_CACHE_SIZE_LIMIT", 1024**3)
-    )
+    cache_size_limit = float(os.environ.get("GLM_BENCHMARKS_CACHE_SIZE_LIMIT", 1024**3))
 
     if force or _get_size_of_cache_directory() > cache_size_limit:
         shutil.rmtree(cache_location)