From 92ffda4b8b010d66ed876ffd943900a84d4e5dc2 Mon Sep 17 00:00:00 2001 From: Andreas Koepf Date: Fri, 14 Feb 2025 12:06:44 +0100 Subject: [PATCH] lazy import of wordle_words --- reasoning_gym/cognition/figlet_fonts.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/reasoning_gym/cognition/figlet_fonts.py b/reasoning_gym/cognition/figlet_fonts.py index 22e81256..7274a4ff 100644 --- a/reasoning_gym/cognition/figlet_fonts.py +++ b/reasoning_gym/cognition/figlet_fonts.py @@ -4,7 +4,6 @@ from typing import Dict, Optional import pyfiglet -from ..data.wordle_words import wordle_words from ..factory import ProceduralDataset, register_dataset @@ -23,6 +22,9 @@ class FigletFontDataset(ProceduralDataset): """Generates FigletFont tasks""" def __init__(self, config: FigletFontConfig): + from ..data.wordle_words import wordle_words + + self.wordle_words = wordle_words self._prompt_templates = [ "What word does this say?\n\n{figlet_render}", "Please read the following figlet font:\n\n{figlet_render}", @@ -40,7 +42,7 @@ class FigletFontDataset(ProceduralDataset): """ rng = Random(self.seed + idx) - word = self.config.static_word if self.config.static_word is not None else rng.choice(wordle_words).upper() + word = self.config.static_word if self.config.static_word is not None else rng.choice(self.wordle_words).upper() if self.config.space_letters: render_word = " ".join(word) else: