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()), ) )