From e2f3ad874894b5a8f021ec9f1ac195456a98e256 Mon Sep 17 00:00:00 2001
From: Sofie Van Landeghem <svlandeg@users.noreply.github.com>
Date: Mon, 11 Dec 2023 15:13:04 +0100
Subject: [PATCH] Fix doc string

---
 spacy_llm/cache.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/spacy_llm/cache.py b/spacy_llm/cache.py
index 92f4e80b..3e1559e1 100644
--- 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):