Backport PR #3126 on branch 1.2.x (misc: Fix the batch size warning i…

…n case of small last batch when using external indices) (#3127) Backport PR #3126: misc: Fix the batch size warning in case of small last batch when using external indices Co-authored-by: Ori Kronfeld <[email protected]>
scverse · Jan 7, 2025 · 32b0b5c · 32b0b5c
1 parent e73e485
commit 32b0b5c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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,