fixed Sequence type call to random.sample

AstraZeneca · Sep 11, 2024 · 1c297c1 · 1c297c1
1 parent d879df3
commit 1c297c1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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()),
                     )
                 )