Fix black

awslabs · Oct 13, 2023 · 580d92d · 580d92d
1 parent 986f94e
commit 580d92d
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/gluonts/dataset/repository/_lstnet.py b/src/gluonts/dataset/repository/_lstnet.py
@@ -141,7 +141,10 @@ def generate_lstnet_dataset(
         pd.read_csv(ds_info.url, header=None),  # type: ignore
     )
 
-    assert df.shape == (ds_info.num_time_steps, ds_info.num_series,), (
+    assert df.shape == (
+        ds_info.num_time_steps,
+        ds_info.num_series,
+    ), (
         "expected num_time_steps/num_series"
         f" {(ds_info.num_time_steps, ds_info.num_series)} but got {df.shape}"
     )