[Ponicode] Docstrings created (#20)

* [ponicode-pull-request] Ponicode wrote new docstrings!

* Update dataset.py

Co-authored-by: github-actions[bot] <41898282+github-actions[bot]@users.noreply.github.com>
Co-authored-by: Jules Belveze <32683010+JulesBelveze@users.noreply.github.com>

Author: github-actions[bot]

tsa/dataset.py

@@ -88,6 +88,9 @@ def get_loaders(self, batch_size: int):
         return train_iter, test_iter, nb_features
 
     def invert_scale(self, predictions):
+        """
+        Inverts the scale of the predictions
+        """
         if isinstance(predictions, torch.Tensor):
             predictions = predictions.numpy()
         unscaled = self.y_scaler.inverse_transform(predictions)