diff --git a/djangoldp_notification/management/commands/mock_notification.py b/djangoldp_notification/management/commands/mock_notification.py index d534a2e..507a3f6 100644 --- a/djangoldp_notification/management/commands/mock_notification.py +++ b/djangoldp_notification/management/commands/mock_notification.py @@ -8,7 +8,5 @@ class Command(BaseCommand): parser.add_argument('--size', type=int, default=0, help='Number of notifications to create') def handle(self, *args, **options): - for i in range(0, options['size']): - notif = NotificationFactory.create() - + NotificationFactory.create_batch(options['size']) self.stdout.write(self.style.SUCCESS('Successful data mock install'))