diff options
Diffstat (limited to 'tests/fixtures/models.py')
| -rw-r--r-- | tests/fixtures/models.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tests/fixtures/models.py b/tests/fixtures/models.py index 03bd8a2..bdb4e12 100644 --- a/tests/fixtures/models.py +++ b/tests/fixtures/models.py @@ -242,7 +242,7 @@ def assignment_record( @pytest.fixture -def assignment_factory(hit: Hit) -> Callable[[Optional[str]], Assignment]: +def assignment_factory(hit_record: Hit) -> Callable[[Optional[str]], Assignment]: def _inner(amt_worker_id: Optional[str] = None) -> Assignment: now = datetime.now(tz=timezone.utc) @@ -251,7 +251,7 @@ def assignment_factory(hit: Hit) -> Callable[[Optional[str]], Assignment]: return Assignment( amt_assignment_id=amt_assignment_id, - amt_hit_id=hit.amt_hit_id, + amt_hit_id=hit_record.amt_hit_id, amt_worker_id=amt_worker_id, status=AssignmentStatus.Submitted, modified_at=now, |