Skip to content
This repository has been archived by the owner on Oct 9, 2023. It is now read-only.

Commit

Permalink
<BLANKLINE>
Browse files Browse the repository at this point in the history
...
  • Loading branch information
Borda committed Aug 31, 2023
1 parent 0f97a93 commit 28ccb03
Show file tree
Hide file tree
Showing 5 changed files with 12 additions and 6 deletions.
3 changes: 2 additions & 1 deletion src/flash/audio/speech_recognition/data.py
Original file line number Diff line number Diff line change
Expand Up @@ -408,7 +408,8 @@ def from_json(
... predict_file="predict_data.json",
... batch_size=2,
... ) # doctest: +ELLIPSIS +NORMALIZE_WHITESPACE
Downloading...
<BLANKLINE>
...
>>> model = SpeechRecognition(backbone="patrickvonplaten/wav2vec2_tiny_random_robust")
>>> trainer = Trainer(fast_dev_run=True)
>>> trainer.fit(model, datamodule=datamodule) # doctest: +ELLIPSIS +NORMALIZE_WHITESPACE
Expand Down
6 changes: 4 additions & 2 deletions src/flash/text/classification/data.py
Original file line number Diff line number Diff line change
Expand Up @@ -316,7 +316,8 @@ def from_json(
... predict_file="predict_data.json",
... batch_size=2,
... ) # doctest: +ELLIPSIS +NORMALIZE_WHITESPACE
Downloading...
<BLANKLINE>
...
>>> datamodule.num_classes
3
>>> datamodule.labels
Expand Down Expand Up @@ -439,7 +440,8 @@ def from_parquet(
... predict_file="predict_data.parquet",
... batch_size=2,
... ) # doctest: +ELLIPSIS +NORMALIZE_WHITESPACE
Downloading...
<BLANKLINE>
...
>>> datamodule.num_classes
3
>>> datamodule.labels
Expand Down
3 changes: 2 additions & 1 deletion src/flash/text/question_answering/data.py
Original file line number Diff line number Diff line change
Expand Up @@ -339,7 +339,8 @@ def from_json(
... predict_file="predict_data.json",
... batch_size=2,
... ) # doctest: +ELLIPSIS +NORMALIZE_WHITESPACE
Downloading...
<BLANKLINE>
...
>>> model = QuestionAnsweringTask(max_source_length=32, max_target_length=32)
>>> trainer = Trainer(fast_dev_run=True)
>>> trainer.fit(model, datamodule=datamodule) # doctest: +ELLIPSIS +NORMALIZE_WHITESPACE
Expand Down
3 changes: 2 additions & 1 deletion src/flash/text/seq2seq/summarization/data.py
Original file line number Diff line number Diff line change
Expand Up @@ -287,7 +287,8 @@ def from_json(
... predict_file="predict_data.json",
... batch_size=2,
... ) # doctest: +ELLIPSIS +NORMALIZE_WHITESPACE
Downloading...
<BLANKLINE>
...
>>> model = SummarizationTask(backbone="JulesBelveze/t5-small-headline-generator")
>>> trainer = Trainer(fast_dev_run=True)
>>> trainer.fit(model, datamodule=datamodule) # doctest: +ELLIPSIS +NORMALIZE_WHITESPACE
Expand Down
3 changes: 2 additions & 1 deletion src/flash/text/seq2seq/translation/data.py
Original file line number Diff line number Diff line change
Expand Up @@ -280,7 +280,8 @@ def from_json(
... predict_file="predict_data.json",
... batch_size=2,
... ) # doctest: +ELLIPSIS +NORMALIZE_WHITESPACE
Downloading...
<BLANKLINE>
...
>>> model = TranslationTask()
>>> trainer = Trainer(fast_dev_run=True)
>>> trainer.fit(model, datamodule=datamodule) # doctest: +ELLIPSIS +NORMALIZE_WHITESPACE
Expand Down

0 comments on commit 28ccb03

Please sign in to comment.