From a6deb6bf59f4475d061c4e96f70696506ade85ab Mon Sep 17 00:00:00 2001 From: Ori Kronfeld Date: Tue, 7 Jan 2025 14:42:59 +0200 Subject: [PATCH] misc: Fix the batch size warning in case of small last batch when using external indices (#3126) --- src/scvi/dataloaders/_data_splitting.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/scvi/dataloaders/_data_splitting.py b/src/scvi/dataloaders/_data_splitting.py index be7c557468..b4b54d8c7f 100644 --- a/src/scvi/dataloaders/_data_splitting.py +++ b/src/scvi/dataloaders/_data_splitting.py @@ -170,7 +170,7 @@ def validate_data_split_with_external_indexing( warnings.warn( f"Last batch will have a small size of {num_of_cells} " f"samples. Consider changing settings.batch_size or batch_size in model.train " - f"from currently {settings.batch_size} to avoid errors during model training " + f"from currently {batch_size} to avoid errors during model training " f"or change the given external indices accordingly.", UserWarning, stacklevel=settings.warnings_stacklevel,