Skip to content

Commit f06aab4

Browse files
committed
Remove python 3.6 uncompatible code
1 parent a67c8a5 commit f06aab4

File tree

1 file changed

+2
-13
lines changed

1 file changed

+2
-13
lines changed

tests/factories.py

Lines changed: 2 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -1,27 +1,16 @@
1-
from typing import Generic, TypeVar
2-
31
import factory
42
import waffle
53
from django.contrib.auth import get_user_model
64

7-
T = TypeVar("T")
85
FlagModel = waffle.get_waffle_flag_model()
96
UserModel = get_user_model()
107

118

12-
class BaseFactory(Generic[T], factory.django.DjangoModelFactory):
13-
"""This is a workaround to get type hints to play nice with factory boy."""
14-
15-
@classmethod
16-
def create(cls, **kwargs) -> T:
17-
return super().create(**kwargs)
18-
19-
20-
class FlagFactory(BaseFactory[FlagModel]):
9+
class FlagFactory(factory.django.DjangoModelFactory):
2110
class Meta:
2211
model = FlagModel
2312

2413

25-
class UserFactory(BaseFactory[UserModel]):
14+
class UserFactory(factory.django.DjangoModelFactory):
2615
class Meta:
2716
model = UserModel

0 commit comments

Comments
 (0)