From 01f10d207d5188efc288ce6c126045b1beeed059 Mon Sep 17 00:00:00 2001 From: Nikita Sokovnin Date: Wed, 6 Mar 2024 22:06:28 +0000 Subject: [PATCH] fix: add future annotations import to synonym generators --- datadreamer/prompt_generation/lm_synonym_generator.py | 2 ++ datadreamer/prompt_generation/wordnet_synonym_generator.py | 2 ++ 2 files changed, 4 insertions(+) diff --git a/datadreamer/prompt_generation/lm_synonym_generator.py b/datadreamer/prompt_generation/lm_synonym_generator.py index fe1d7c6..e1c4401 100644 --- a/datadreamer/prompt_generation/lm_synonym_generator.py +++ b/datadreamer/prompt_generation/lm_synonym_generator.py @@ -1,3 +1,5 @@ +from __future__ import annotations + import re from typing import List, Optional diff --git a/datadreamer/prompt_generation/wordnet_synonym_generator.py b/datadreamer/prompt_generation/wordnet_synonym_generator.py index 1e96f8f..4f8f6f8 100644 --- a/datadreamer/prompt_generation/wordnet_synonym_generator.py +++ b/datadreamer/prompt_generation/wordnet_synonym_generator.py @@ -1,3 +1,5 @@ +from __future__ import annotations + from typing import List, Optional import nltk