From fc505f5072ec25debc08b676460e782b006f0924 Mon Sep 17 00:00:00 2001 From: Shivam Sandbhor Date: Sat, 6 Jun 2020 10:14:06 +0530 Subject: [PATCH] Add type conversion to int, when batch_size is specified via cli Signed-off-by: Shivam Sandbhor --- vulnerabilities/management/commands/import.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/vulnerabilities/management/commands/import.py b/vulnerabilities/management/commands/import.py index 6a8c5eacd..0991f596e 100644 --- a/vulnerabilities/management/commands/import.py +++ b/vulnerabilities/management/commands/import.py @@ -100,7 +100,7 @@ def import_data(self, names, cutoff_date): def _import_data(self, importers, cutoff_date): for importer in importers: self.stdout.write(f'Importing data from {importer.name}') - batch_size = getattr(self, 'batch_size', 10) + batch_size = int(getattr(self, 'batch_size', 10)) ImportRunner(importer, batch_size).run(cutoff_date=cutoff_date) self.stdout.write( self.style.SUCCESS(f'Successfully imported data from {importer.name}'))