diff --git a/CHANGELOG.md b/CHANGELOG.md index 56123f3133..f63aa56e74 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -20,6 +20,8 @@ The format is based on [Keep a Changelog](http://keepachangelog.com/en/1.0.0/). ### Fixed +- Fixed examples (question answering), where NLTK's `punkt` module needs to be downloaded first. ([#1215](https://github.com/PyTorchLightning/lightning-flash/pull/1215/files)) + - Fixed a bug where DDP would not work with Flash tasks ([#1182](https://github.com/PyTorchLightning/lightning-flash/pull/1182)) - Fixed DDP support for `VideoClassifier` ([#1189](https://github.com/PyTorchLightning/lightning-flash/pull/1189)) diff --git a/flash_examples/question_answering.py b/flash_examples/question_answering.py index 0306efe92f..c1a2306897 100644 --- a/flash_examples/question_answering.py +++ b/flash_examples/question_answering.py @@ -13,8 +13,15 @@ # limitations under the License. from flash import Trainer from flash.core.data.utils import download_data +from flash.core.utilities.imports import example_requires from flash.text import QuestionAnsweringData, QuestionAnsweringTask +example_requires("text") + +import nltk # noqa: E402 + +nltk.download("punkt") + # 1. Create the DataModule download_data("https://pl-flash-data.s3.amazonaws.com/squad_tiny.zip", "./data/")