Skip to content

Commit c1a661a

Browse files
committed
use self_type when generating managers
1 parent c203556 commit c1a661a

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

mypy_django_plugin/transformers/models.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -343,7 +343,8 @@ def run_with_model_cls(self, model_cls: Type[Model]) -> None:
343343
incomplete_manager_defs.add(manager_name)
344344
continue
345345

346-
manager_type = Instance(manager_info, [Instance(self.model_classdef.info, [])])
346+
assert self.model_classdef.info.self_type is not None
347+
manager_type = Instance(manager_info, [self.model_classdef.info.self_type])
347348
self.add_new_node_to_model_class(manager_name, manager_type, is_classvar=True)
348349

349350
if incomplete_manager_defs:

0 commit comments

Comments
 (0)