Fix doc string

explosion · Dec 11, 2023 · e2f3ad8 · e2f3ad8
1 parent 63367fa
commit e2f3ad8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/spacy_llm/cache.py b/spacy_llm/cache.py
@@ -72,7 +72,7 @@ def initialize(self, vocab: Vocab, task: LLMTask) -> None:
         """
         Initialize cache with data not available at construction time.
         vocab (Vocab): Vocab object.
-        task (ShardingLLMTask): Task.
+        task (LLMTask): Task.
         """
         self._vocab = vocab
         if isinstance(task, PromptTemplateProvider):