File tree Expand file tree Collapse file tree 1 file changed +3
-2
lines changed Expand file tree Collapse file tree 1 file changed +3
-2
lines changed Original file line number Diff line number Diff line change 33from django .db import models
44from django .test import TestCase
55
6+ from model_utils .managers import InheritanceManager
67from tests .models import (
78 InheritanceManagerTestChild1 ,
89 InheritanceManagerTestChild2 ,
@@ -25,7 +26,7 @@ def setUp(self) -> None:
2526 self .grandchild1_2 = \
2627 InheritanceManagerTestGrandChild1_2 .objects .create ()
2728
28- def get_manager (self ):
29+ def get_manager (self ) -> InheritanceManager [ InheritanceManagerTestParent ] :
2930 return InheritanceManagerTestParent .objects
3031
3132 def test_normal (self ) -> None :
@@ -495,7 +496,7 @@ def setUp(self) -> None:
495496 self .grandchild1 = InheritanceManagerTestGrandChild1 .objects .create (related = self .related )
496497 self .grandchild1_2 = InheritanceManagerTestGrandChild1_2 .objects .create (related = self .related )
497498
498- def get_manager (self ):
499+ def get_manager (self ) -> models . Manager :
499500 return self .related .imtests
500501
501502 def test_get_method_with_select_subclasses (self ) -> None :
You can’t perform that action at this time.
0 commit comments