From eab2f966268f08c95d951b824c8f60378c824b8e Mon Sep 17 00:00:00 2001 From: David Chanin Date: Mon, 16 May 2022 02:21:12 +0100 Subject: [PATCH] fixing tests --- frame_semantic_transformer/data/framenet.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/frame_semantic_transformer/data/framenet.py b/frame_semantic_transformer/data/framenet.py index 661c36e..a78c637 100644 --- a/frame_semantic_transformer/data/framenet.py +++ b/frame_semantic_transformer/data/framenet.py @@ -38,7 +38,7 @@ def get_frame_elements_map_by_core_type() -> dict[str, dict[str, list[str]]]: results: dict[str, dict[str, list[str]]] = {} for frame in fn.frames(): results[frame.name] = {"core": [], "noncore": []} - for element_name, element in fn.frame(frame).FE.items(): + for element_name, element in frame.FE.items(): element_type = "core" if element.coreType == "Core" else "noncore" results[frame.name][element_type].append(element_name) return results