fix minor formatting issues

Author: stes

cebra/integrations/sklearn/cebra.py

@@ -28,9 +28,9 @@
 import numpy as np
 import numpy.typing as npt
 import pkg_resources
+import sklearn
 import sklearn.utils.validation as sklearn_utils_validation
 import torch
-import sklearn
 from sklearn.base import BaseEstimator
 from sklearn.base import TransformerMixin
 from sklearn.utils.metaestimators import available_if
@@ -43,12 +43,14 @@
 import cebra.models
 import cebra.solver
 
+
 def check_version(estimator):
     # NOTE(stes): required as a check for the old way of specifying tags
     # https://github.com/scikit-learn/scikit-learn/pull/29677#issuecomment-2334229165
     from packaging import version
     return version.parse(sklearn.__version__) < version.parse("1.6.dev")
 
+
 def _init_loader(
     is_cont: bool,
     is_disc: bool,

docs/source/conf.py

@@ -28,18 +28,13 @@
 
 # -- Path setup --------------------------------------------------------------
 
-# If extensions (or modules to document with autodoc) are in another directory,
-# add these directories to sys.path here. If the directory is relative to the
-# documentation root, use os.path.abspath to make it absolute, like shown here.
-#
+import datetime
 import os
 import sys
 
 sys.path.insert(0, os.path.abspath("."))
 
-import datetime
-
-import cebra
+import cebra  # noqa: E402
 
 
 def get_years(start_year=2021):
@@ -247,6 +242,9 @@ def get_years(start_year=2021):
 
 # Download link for the notebook, see
 # https://nbsphinx.readthedocs.io/en/0.3.0/prolog-and-epilog.html
+
+# fmt: off
+# flake8: noqa: E501
 nbsphinx_prolog = r"""
 
 .. only:: html
@@ -269,3 +267,5 @@ def get_years(start_year=2021):
 
 ----
 """
+# fmt: on
+# flake8: enable=E501