[BUG] pytorch-forecasting#1752 Fixing

Author: RUPESH-KUMAR01

pytorch_forecasting/data/timeseries.py

@@ -2513,7 +2513,7 @@ def to_dataloader(
 
         Parameters
         ----------
-        train : bool, optional, default=Trze
+        train : bool, optional, default=True
             whether dataloader is used for training (True) or prediction (False).
             Will shuffle and drop last batch if True. Defaults to True.
         batch_size : int, optional, default=64

pytorch_forecasting/models/base/_base_model.py

@@ -387,7 +387,7 @@ def on_predict_epoch_end(
             if self.return_decoder_lengths:
                 output["decoder_lengths"] = torch.cat(self._decode_lengths, dim=0)
             if self.return_y:
-                y = concat_sequences([yi[0] for yi in self._y])
+                y = _torch_cat_na([yi[0] for yi in self._y])
                 if self._y[-1][1] is None:
                     weight = None
                 else: