From f70a23c076eeb7bc1df005267744419f836f757d Mon Sep 17 00:00:00 2001 From: Dario Coscia Date: Mon, 17 Mar 2025 13:47:08 +0100 Subject: [PATCH] formatting --- pina/data/data_module.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pina/data/data_module.py b/pina/data/data_module.py index a70a001..6f3c751 100644 --- a/pina/data/data_module.py +++ b/pina/data/data_module.py @@ -278,7 +278,7 @@ class PinaDataModule(LightningDataModule): be in the range [0, 1]. :param float val_size: Fraction of elements in the validation split. It must be in the range [0, 1]. - :param int batch_size: The batch size used for training. If ``None``, + :param int batch_size: The batch size used for training. If ``None``, the entire dataset is returned in a single batch. Default is ``None``. :param bool shuffle: Whether to shuffle the dataset before splitting.