Skip to content

Commit

Permalink
refactor: rename pipeline to not hit test
Browse files Browse the repository at this point in the history
  • Loading branch information
MartinBernstorff committed Nov 18, 2023
1 parent b81c54a commit 1b5cbf2
Showing 1 changed file with 3 additions and 3 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -31,7 +31,7 @@
)


class TestCardPipeline(CardPipeline):
class MockCardPipeline(CardPipeline):
def __init__(
self,
document_factory: DocumentFactory = MarkdownNoteFactory(), # noqa: B008
Expand Down Expand Up @@ -96,7 +96,7 @@ def test_qa_uuid_generation():
/ "test_md_files"
/ "test_card_guid.md"
)
cards = TestCardPipeline(
cards = MockCardPipeline(
prompt_extractors=[QAPromptExtractor()]
).run(input_path=file_path)
notes = [c.to_genanki_note() for c in cards]
Expand All @@ -114,7 +114,7 @@ def test_cloze_uuid_generation():
/ "test_md_files"
/ "test_card_guid.md"
)
cloze_cards = TestCardPipeline(
cloze_cards = MockCardPipeline(
prompt_extractors=[ClozePromptExtractor()]
).run(input_path=file_path)

Expand Down

0 comments on commit 1b5cbf2

Please sign in to comment.