diff --git a/generate_pun.py b/generate_pun.py index fa0d49c..5c710c1 100644 --- a/generate_pun.py +++ b/generate_pun.py @@ -5,6 +5,7 @@ from collections import defaultdict import fuzzy + from fairseq import options from pungen.retriever import Retriever