diff --git a/django-stubs/db/models/query.pyi b/django-stubs/db/models/query.pyi index 670927f..de51553 100644 --- a/django-stubs/db/models/query.pyi +++ b/django-stubs/db/models/query.pyi @@ -84,9 +84,7 @@ class QuerySet(Iterable[_T], Sized): def latest(self, *fields: Any, field_name: Optional[Any] = ...) -> _T: ... def first(self) -> Optional[_T]: ... def last(self) -> Optional[_T]: ... - def in_bulk( - self, id_list: Any = ..., *, field_name: str = ..., **kwargs: Any - ) -> Dict[Union[int, str], _T]: ... + def in_bulk(self, id_list: Sequence[Any] = ..., *, field_name: str = ...) -> Dict[Any, _T]: ... def delete(self) -> Tuple[int, Dict[str, int]]: ... def update(self, **kwargs: Any) -> int: ... def _update(self, values: Any) -> Optional[Any]: ... diff --git a/test-data/typecheck/queryset.test b/test-data/typecheck/queryset.test new file mode 100644 index 0000000..25882af --- /dev/null +++ b/test-data/typecheck/queryset.test @@ -0,0 +1,10 @@ +[CASE test_queryset] +from django.db import models + +class Blog(models.Model): + slug = models.CharField(max_length=100) +reveal_type(Blog.objects.in_bulk([1])) # E: Revealed type is 'builtins.dict[Any, main.Blog*]' +reveal_type(Blog.objects.in_bulk()) # E: Revealed type is 'builtins.dict[Any, main.Blog*]' +reveal_type(Blog.objects.in_bulk(['beatles_blog'], field_name='slug')) # E: Revealed type is 'builtins.dict[Any, main.Blog*]' + +[out]