diff --git a/tests/unit/test_confirm_generation.py b/tests/unit/test_confirm_generation.py index 2ab258af4..72f70856d 100644 --- a/tests/unit/test_confirm_generation.py +++ b/tests/unit/test_confirm_generation.py @@ -60,10 +60,13 @@ class StubGenerationTaskRepository: return len([t for t in self._store.values() if t.created_by_user_id == user_id]) def count_pending_by_user(self, user_id: str) -> int: - return len([ - t for t in self._store.values() - if t.created_by_user_id == user_id and t.status == GenerationTaskStatus.PENDING - ]) + return len( + [ + t + for t in self._store.values() + if t.created_by_user_id == user_id and t.status == GenerationTaskStatus.PENDING + ] + ) def count_pending_total(self) -> int: return len([t for t in self._store.values() if t.status == GenerationTaskStatus.PENDING])