Skip to content

GitLab

Commit c179f7af authored by Baber's avatar Baber
Browse files

add cache

parent 134b4865
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
lm_eval/tasks/ruler/essays.py
View file @ c179f7af
...@@ -15,6 +15,7 @@ import asyncio ...@@ -15,6 +15,7 @@ import asyncio
import glob import glob
import os import os
import shutil import shutil
from functools import cache
from typing import Dict from typing import Dict
import html2text import html2text
...@@ -118,6 +119,7 @@ async def get_essays() -> Dict[str, str]: ...@@ -118,6 +119,7 @@ async def get_essays() -> Dict[str, str]:
return {"text": text} return {"text": text}
@cache
def get_all_essays() -> Dict[str, str]: def get_all_essays() -> Dict[str, str]:
"""Synchronous wrapper for get_essays()""" """Synchronous wrapper for get_essays()"""
return asyncio.run(get_essays()) return asyncio.run(get_essays())
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment