lint: freeze & run Black version 24.02

.github/workflows/style_type_checks.yml

@@ -9,22 +9,16 @@ jobs:
     steps:
     - uses: actions/checkout@v3
     - uses: extractions/setup-just@v1
-      env:
-        GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
-    - name: Set up Python 3.8
-      uses: actions/setup-python@v4
-      with:
-        python-version: '3.8'
+    - uses: actions/setup-python@v4
     - name: Install dependencies
       run: |
         pip install .
-        pip install click black mypy
-        pip install types-python-dateutil
-        pip install types-waitress
-        pip install types-PyYAML
-    - name: Style and type checks
-      run: |
-        just black
-        just mypy
+        # todo: install also `black[jupyter]`
+        pip install click "black==24.02" "mypy==1.8.0" \
+            types-python-dateutil types-waitress types-PyYAML
+    - name: Style check
+      run: just black
+    - name: Type check
+      run: just mypy
     - name: Check license headers
       run: just license

Justfile

@@ -34,7 +34,7 @@ release:
   python setup.py sdist
 
 black:
-  black --check src test examples
+  black --check --color src test examples
 
 mypy:
   python setup.py type_check

examples/anomaly_detection.py

@@ -14,6 +14,7 @@
 """
 This example shows how to do anomaly detection with DeepAR.
 The model is first trained and then time-points with the largest negative log-likelihood are plotted.
+
 """
 import numpy as np
 from itertools import islice

examples/warm_start.py

@@ -13,6 +13,7 @@
 
 """
 This example show how to intialize the network with parameters from a model that was previously trained.
+
 """
 
 from gluonts.dataset.repository import get_dataset, dataset_recipes

pyproject.toml

@@ -1,4 +1,5 @@
 [tool.black]
+target-version = ['py38']
 line-length = 79
 
 [tool.pytest.ini_options]

src/gluonts/ext/rotbaum/_model.py

@@ -142,7 +142,7 @@ def fit(
         # XGBoost, but True if one uses lightgbm.
         model_is_already_trained: bool = False,  # True if there is no need to
         # train self.model
-        **kwargs
+        **kwargs,
     ):
         """
         Fits self.model and partitions R^n into cells.

src/gluonts/ext/rotbaum/_preprocess.py

@@ -49,7 +49,7 @@ def __init__(
         max_n_datapts: int = 400000,
         seed: Optional[int] = None,
         num_samples: Optional[int] = None,
-        **kwargs
+        **kwargs,
     ):
         """
         Parameters
@@ -296,7 +296,7 @@ def __init__(
         one_hot_encode: bool = True,
         subtract_mean: bool = True,
         count_nans: bool = False,
-        **kwargs
+        **kwargs,
     ):
         if one_hot_encode:
             assert cardinality != "ignore" or (
@@ -313,7 +313,7 @@ def __init__(
             stratify_targets=stratify_targets,
             n_ignore_last=n_ignore_last,
             num_samples=num_samples,
-            **kwargs
+            **kwargs,
         )
 
         self.use_feat_static_real = use_feat_static_real

src/gluonts/model/evaluation.py

@@ -104,7 +104,7 @@ def evaluate_forecasts_raw(
     batch_size: int = 100,
     mask_invalid_label: bool = True,
     allow_nan_forecast: bool = False,
-    seasonality: Optional[int] = None
+    seasonality: Optional[int] = None,
 ) -> dict:
     """
     Evaluate ``forecasts`` by comparing them with ``test_data``, according
@@ -189,7 +189,7 @@ def evaluate_forecasts(
     batch_size: int = 100,
     mask_invalid_label: bool = True,
     allow_nan_forecast: bool = False,
-    seasonality: Optional[int] = None
+    seasonality: Optional[int] = None,
 ) -> pd.DataFrame:
     """
     Evaluate ``forecasts`` by comparing them with ``test_data``, according
@@ -243,7 +243,7 @@ def evaluate_model(
     batch_size: int = 100,
     mask_invalid_label: bool = True,
     allow_nan_forecast: bool = False,
-    seasonality: Optional[int] = None
+    seasonality: Optional[int] = None,
 ) -> pd.DataFrame:
     """
     Evaluate ``model`` when applied to ``test_data``, according

src/gluonts/model/forecast_generator.py

@@ -94,7 +94,7 @@ def __call__(
         input_names: List[str],
         output_transform: Optional[OutputTransform],
         num_samples: Optional[int],
-        **kwargs
+        **kwargs,
     ) -> Iterator[Forecast]:
         raise NotImplementedError()
 
@@ -111,7 +111,7 @@ def __call__(
         input_names: List[str],
         output_transform: Optional[OutputTransform],
         num_samples: Optional[int],
-        **kwargs
+        **kwargs,
     ) -> Iterator[Forecast]:
         for batch in inference_data_loader:
             inputs = select(input_names, batch, ignore_missing=True)
@@ -155,7 +155,7 @@ def __call__(
         input_names: List[str],
         output_transform: Optional[OutputTransform],
         num_samples: Optional[int],
-        **kwargs
+        **kwargs,
     ) -> Iterator[Forecast]:
         for batch in inference_data_loader:
             inputs = select(input_names, batch, ignore_missing=True)
@@ -205,7 +205,7 @@ def __call__(
         input_names: List[str],
         output_transform: Optional[OutputTransform],
         num_samples: Optional[int],
-        **kwargs
+        **kwargs,
     ) -> Iterator[Forecast]:
         for batch in inference_data_loader:
             inputs = select(input_names, batch, ignore_missing=True)

src/gluonts/mx/block/regularization.py

@@ -49,7 +49,7 @@ def __init__(
         weight: Optional[float] = None,
         batch_axis: int = 1,
         time_axis: int = 0,
-        **kwargs
+        **kwargs,
     ):
         super().__init__(weight, batch_axis, **kwargs)
         self._alpha = alpha
@@ -121,7 +121,7 @@ def __init__(
         weight: Optional[float] = None,
         batch_axis: int = 1,
         time_axis: int = 0,
-        **kwargs
+        **kwargs,
     ):
         super().__init__(weight, batch_axis, **kwargs)
         self._beta = beta

src/gluonts/mx/block/scaler.py

@@ -121,7 +121,7 @@ def __init__(
         minimum_scale: float = 1e-10,
         default_scale: Optional[float] = None,
         *args,
-        **kwargs
+        **kwargs,
     ):
         super().__init__(*args, **kwargs)
         self.minimum_scale = minimum_scale

src/gluonts/mx/block/sndense.py

@@ -51,7 +51,7 @@ def __init__(
         dtype="float32",
         num_power_iter: int = 1,
         ctx: Optional[mx.Context] = None,
-        **kwargs
+        **kwargs,
     ):
         super().__init__(**kwargs)
         self._coeff = coeff

src/gluonts/mx/distribution/iresnet.py

@@ -65,7 +65,7 @@ def __init__(
         coeff: float = 0.9,
         use_caching: bool = True,
         *args,
-        **kwargs
+        **kwargs,
     ):
         super().__init__(*args, **kwargs)
         assert len(event_shape) == 1

src/gluonts/mx/distribution/lds.py

@@ -65,7 +65,7 @@ def _safe_split(x, num_outputs, axis, squeeze_axis, *args, **kwargs):
             num_outputs=num_outputs,
             squeeze_axis=squeeze_axis,
             *args,
-            **kwargs
+            **kwargs,
         )
     return [x.squeeze(axis=axis)] if squeeze_axis else [x]
 

src/gluonts/nursery/SCott/dataset_tools/algo_clustering.py

@@ -232,14 +232,14 @@ def KMeans_m5_dataset(
             {
                 "target": ts_slice,
                 "start": unsplit_start,
-            }  # , 'feat_static_cat': train_entry['feat_static_cat']}
-        )
+            }
+        )  # , 'feat_static_cat': train_entry['feat_static_cat']}
         whole_data.append(
             {
                 "target": ts_slice,
                 "start": unsplit_start,
-            }  # , 'feat_static_cat': train_entry['feat_static_cat']}
-        )
+            }
+        )  # , 'feat_static_cat': train_entry['feat_static_cat']}
         sample_id += 1
     print(len(whole_data))
     ret["group_ratio"] = [len(i) / len(whole_data) for i in dataset_group]

src/gluonts/nursery/daf/tslib/nn/attention/selfattn.py

@@ -309,7 +309,7 @@ def forward(
         value: Tensor,
         shape: Tensor,
         *,
-        mask: Optional[BoolTensor] = None
+        mask: Optional[BoolTensor] = None,
     ) -> Tensor:
         q, k, v = self._compute_qkv(value, shape)
         score = self._compute_attn_score(q, k, mask)

src/gluonts/nursery/robust-mts-attack/pts/modules/gaussian_diffusion.py

@@ -312,7 +312,7 @@ def log_prob(self, x, cond, *args, **kwargs):
             cond.reshape(B * T, 1, -1),
             time,
             *args,
-            **kwargs
+            **kwargs,
         )
 
         return loss

src/gluonts/nursery/temporal_hierarchical_forecasting/model/cop_deepar/gnn.py

@@ -23,7 +23,7 @@ def __init__(
         num_layers: int,
         adj_matrix: Tensor,
         use_mlp: bool = True,
-        **kwargs
+        **kwargs,
     ):
         super().__init__(**kwargs)
 

src/gluonts/nursery/tsbench/src/cli/analysis/scripts/ensemble_recommender.py

@@ -100,7 +100,7 @@ def main(
             surrogate[surrogate["name"]]
             if surrogate["name"] in surrogate
             else {}
-        )
+        ),
     )
 
     # Then, we can create the recommender

src/gluonts/nursery/tsbench/src/cli/analysis/scripts/recommender.py

@@ -123,7 +123,7 @@ def main(
                 surrogate[surrogate["name"]]
                 if surrogate["name"] in surrogate
                 else {}
-            )
+            ),
         )
     elif recommender == "optimal":
         recommender_args["tracker"] = tracker

src/gluonts/nursery/tsbench/src/cli/analysis/scripts/surrogate.py

@@ -92,7 +92,7 @@ def main(
         input_flags=inputs,
         output_normalization=outputs["normalization"],
         impute_simulatable=outputs["imputation"],
-        **(_config[surrogate] if surrogate in _config else {})
+        **(_config[surrogate] if surrogate in _config else {}),
     )
 
     # And evaluate it

src/gluonts/torch/model/i_transformer/estimator.py

@@ -232,7 +232,7 @@ def create_training_data_loader(
         data: Dataset,
         module: ITransformerLightningModule,
         shuffle_buffer_length: Optional[int] = None,
-        **kwargs
+        **kwargs,
     ) -> Iterable:
         data = Cyclic(data).stream()
         instances = self._create_instance_splitter(module, "training").apply(

src/gluonts/torch/model/lag_tst/estimator.py

@@ -225,7 +225,7 @@ def create_training_data_loader(
         data: Dataset,
         module: LagTSTLightningModule,
         shuffle_buffer_length: Optional[int] = None,
-        **kwargs
+        **kwargs,
     ) -> Iterable:
         data = Cyclic(data).stream()
         instances = self._create_instance_splitter(module, "training").apply(

src/gluonts/torch/model/patch_tst/estimator.py

@@ -229,7 +229,7 @@ def create_training_data_loader(
         data: Dataset,
         module: PatchTSTLightningModule,
         shuffle_buffer_length: Optional[int] = None,
-        **kwargs
+        **kwargs,
     ) -> Iterable:
         data = Cyclic(data).stream()
         instances = self._create_instance_splitter(module, "training").apply(

src/gluonts/torch/model/wavenet/estimator.py

@@ -348,7 +348,7 @@ def create_training_data_loader(
         data: Dataset,
         module: WaveNetLightningModule,
         shuffle_buffer_length: Optional[int] = None,
-        **kwargs
+        **kwargs,
     ) -> Iterable:
         data = Cyclic(data).stream()
         instances = self._create_instance_splitter("training").apply(

test/nursery/sagemaker_sdk/test_entry_point_scripts.py

@@ -91,7 +91,7 @@ def test_train_script(dataset_name, custom_dataset):
         estimator = estimator_cls.from_hyperparameters(
             prediction_length=dataset.metadata.prediction_length,
             freq=dataset.metadata.freq,
-            **hyperparameters
+            **hyperparameters,
         )
         serialized = serde.dump_json(estimator)
         with open(temp_dir_path / "estimator.json", "w") as estimator_file: