fix typo

Signed-off-by: Igor Davidyuk <igor.davidyuk@intel.com>
openvinotoolkit · May 29, 2024 · 1b87369 · 1b87369
1 parent 3bd00bc
commit 1b87369
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/geti_sdk/data_models/annotation_scene.py b/geti_sdk/data_models/annotation_scene.py
@@ -378,7 +378,7 @@ def filter_annotations(
         :return: AnnotationScene with filtered annotations
         """
         label_names_to_keep = {
-            label if type(label) == str else label.name for label in labels
+            label if type(label) is str else label.name for label in labels
         }
         filtered_annotations: List[Annotation] = []
         for annotation in self.annotations: