diff --git a/tests/test_task_loader.py b/tests/test_task_loader.py index 7ac118d4..0c04f93b 100644 --- a/tests/test_task_loader.py +++ b/tests/test_task_loader.py @@ -70,7 +70,7 @@ def _gen_task_loader_call_args(self, n_context_sets, n_target_sets): Loops over all possible combinations of context/target sampling methods and returns a list of arguments for ``TaskLoader.__call__``. Options tested include: - + - (int): Random number of samples - (float): Fraction of samples - "all": All samples