Skip to content

Commit d5b1a5d

Browse files
Don't need __model__ declaration in builder classes.
1 parent f441269 commit d5b1a5d

File tree

1 file changed

+2
-6
lines changed

1 file changed

+2
-6
lines changed

tests/utils/builders.py

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -7,22 +7,18 @@
77
from eligibility_signposting_api.model.rules import CampaignConfig, Iteration, IterationCohort, IterationRule
88

99

10-
class IterationCohortFactory(ModelFactory[IterationCohort]):
11-
__model__ = IterationCohort
10+
class IterationCohortFactory(ModelFactory[IterationCohort]): ...
1211

1312

14-
class IterationRuleFactory(ModelFactory[IterationRule]):
15-
__model__ = IterationRule
13+
class IterationRuleFactory(ModelFactory[IterationRule]): ...
1614

1715

1816
class IterationFactory(ModelFactory[Iteration]):
19-
__model__ = Iteration
2017
iteration_cohorts = Use(IterationCohortFactory.batch, size=2)
2118
iteration_rules = Use(IterationRuleFactory.batch, size=2)
2219

2320

2421
class CampaignConfigFactory(ModelFactory[CampaignConfig]):
25-
__model__ = CampaignConfig
2622
iterations = Use(IterationFactory.batch, size=2)
2723

2824

0 commit comments

Comments
 (0)