Fix formatting

Author: Jan Beitner

pytorch_forecasting/data/timeseries.py

@@ -1094,7 +1094,8 @@ def __init__(
 
         Args:
             data_source (TimeSeriesDataSet): timeseries dataset.
-            drop_last (bool): if to drop last mini-batch from a group if it is smaller than batch_size. Defaults to False.
+            drop_last (bool): if to drop last mini-batch from a group if it is smaller than batch_size.
+                Defaults to False.
             shuffle (bool): if to shuffle dataset. Defaults to False.
             batch_size (int, optional): Number of samples in a mini-batch. This is rather the maximum number
                 of samples. Because mini-batches are grouped by prediction time, chances are that there

tests/test_data.py

@@ -7,14 +7,12 @@
 import pytest
 from sklearn.preprocessing import StandardScaler
 import torch
-from torch.utils.data import dataset
 
 from pytorch_forecasting.data import (
     EncoderNormalizer,
     GroupNormalizer,
     NaNLabelEncoder,
     TimeSeriesDataSet,
-    TimeSynchronizedBatchSampler,
 )
 from pytorch_forecasting.data.examples import get_stallion_data