From c648e201f394603e5a19ba8b390cda6ac71dbb00 Mon Sep 17 00:00:00 2001 From: Max Woolf Date: Thu, 6 May 2021 22:05:38 -0700 Subject: [PATCH] don't change loop while running --- aitextgen/aitextgen.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/aitextgen/aitextgen.py b/aitextgen/aitextgen.py index f9cff5f..21bcf45 100644 --- a/aitextgen/aitextgen.py +++ b/aitextgen/aitextgen.py @@ -403,9 +403,10 @@ def generate( # remove fields not in schema_return if schema_return: + keys = gen_text_dict.keys() if len(schema_return) == 1: gen_text_dict = gen_text_dict[schema_return[0]] - for key in gen_text_dict.keys(): + for key in keys: if key not in schema_return: gen_text_dict.pop(key, None)