Skip to content

Commit 943fffe

Browse files
authored
fix: UTC-444: Using iterate_queryset in allow_skip migration (#9028)
Co-authored-by: mcanu <mcanu@users.noreply.github.com>
1 parent 89227f4 commit 943fffe

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

label_studio/data_manager/migrations/0018_remove_allow_skip.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -13,6 +13,7 @@
1313
from django.db import migrations
1414

1515
from core.redis import start_job_async_or_sync
16+
from core.utils.iterators import iterate_queryset
1617

1718
logger = logging.getLogger(__name__)
1819
migration_name = __name__
@@ -35,7 +36,7 @@ def forward_migration_job(*, migration_name: str) -> None:
3536
migration.save()
3637

3738
try:
38-
views = View.objects.all()
39+
views = iterate_queryset(View.objects.all())
3940
updated_count = 0
4041

4142
for view in views:
@@ -74,7 +75,7 @@ def reverse_migration_job(*, migration_name: str) -> None:
7475

7576
logger.info(f'Starting reverse migration {migration_name}')
7677

77-
views = View.objects.all()
78+
views = iterate_queryset(View.objects.all())
7879
updated_count = 0
7980

8081
for view in views:

0 commit comments

Comments
 (0)