format code

This commit is contained in:
Dario Coscia
2025-03-17 12:45:16 +01:00
committed by Nicola Demo
parent f34bc55dfd
commit 7cbe658303

View File

@@ -81,7 +81,7 @@ class Collator:
:param dict max_conditions_lengths: ``dict`` containing the maximum :param dict max_conditions_lengths: ``dict`` containing the maximum
number of data points to consider in a single batch for number of data points to consider in a single batch for
each condition. each condition.
:param bool automatic_batching: Whether to enable automatic batching. :param bool automatic_batching: Whether to enable automatic batching.
If ``True``, automatic PyTorch batching If ``True``, automatic PyTorch batching
is performed, which consists of extracting one element at a time is performed, which consists of extracting one element at a time
from the dataset and collating them into a batch. This is useful from the dataset and collating them into a batch. This is useful