remove redundant _examples field (run-llama#265)

* remove redundant _examples field * formatting
viveksilimkhan1 · Aug 17, 2023 · 498826c · 498826c
1 parent 6874426
commit 498826c
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/spacy_llm/tasks/sentiment.py b/spacy_llm/tasks/sentiment.py
@@ -58,7 +58,6 @@ def __init__(
         field (str): The name of the doc extension in which to store the summary.
         """
         self._template = template
-        self._examples = examples
         self._prompt_examples = examples or []
         self._field = field
         self._check_doc_extension()
@@ -100,7 +99,7 @@ def generate_prompts(self, docs: Iterable[Doc]) -> Iterable[str]:
         for doc in docs:
             prompt = _template.render(
                 text=doc.text,
-                examples=self._examples,
+                examples=self._prompt_examples,
             )
             yield prompt
 

diff --git a/spacy_llm/tasks/summarization.py b/spacy_llm/tasks/summarization.py
@@ -65,7 +65,6 @@ def __init__(
         field (str): The name of the doc extension in which to store the summary.
         """
         self._template = template
-        self._examples = examples
         self._max_n_words = max_n_words
         self._field = field
         self._prompt_examples = examples or []
@@ -126,7 +125,9 @@ def generate_prompts(self, docs: Iterable[Doc]) -> Iterable[str]:
         _template = environment.from_string(self._template)
         for doc in docs:
             prompt = _template.render(
-                text=doc.text, examples=self._examples, max_n_words=self._max_n_words
+                text=doc.text,
+                examples=self._prompt_examples,
+                max_n_words=self._max_n_words,
             )
             yield prompt