From 14738e8356a615a9102d7f393f73500e325ee73e Mon Sep 17 00:00:00 2001 From: Richard Jackson Date: Wed, 11 Sep 2024 13:12:43 +0100 Subject: [PATCH] fixed Sequence type call to random.sample --- kazu/steps/ner/gliner.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/kazu/steps/ner/gliner.py b/kazu/steps/ner/gliner.py index dc8719c1..ac77d72e 100644 --- a/kazu/steps/ner/gliner.py +++ b/kazu/steps/ner/gliner.py @@ -225,7 +225,7 @@ def __init__( self.label_sets.add( tuple( random.sample( - self.gliner_class_prompt_to_entity_class.keys(), + list(self.gliner_class_prompt_to_entity_class.keys()), k=len(self.gliner_class_prompt_to_entity_class.keys()), ) )