fixed cache

data.py

@@ -236,18 +236,18 @@ def generate_annotated_images():
         yield AnnotatedImage.from_image_id(image_id)
 
 
-@functools.cache
+@functools.lru_cache
 def load_train_image_ids() -> list[str]:
     train_image_ids = [i.replace(".jpg", "") for i in os.listdir("data/train/images")]
     return train_image_ids[: 1000 if CONFIG.debug else None]
 
 
-@functools.cache
+@functools.lru_cache
 def load_test_image_ids() -> list[str]:
     return [i.replace(".jpg", "") for i in os.listdir("data/test/images")]
 
 
-@functools.cache
+@functools.lru_cache
 def load_image_annotation(image_id: str) -> dict:
     return json.load(open(f"data/train/annotations/{image_id}.json"))
 
@@ -309,7 +309,7 @@ def to_token_str(value: str or enum.Enum):
         return f"<{string}>"
 
 
-@functools.cache
+@functools.lru_cache
 def get_extra_tokens() -> types.SimpleNamespace:
     token_ns = types.SimpleNamespace()