test_fixes

Author: Gabefire

libs/labelbox/src/labelbox/schema/ontology.py

@@ -569,7 +569,7 @@ class OntologyBuilder:
     def from_dict(cls, dictionary: Dict[str, Any]) -> Dict[str, Any]:
         classifications = []
         for c in dictionary["classifications"]:
-            if ["type"] in c and c["type"] in PromptResponseClassification.Type:
+            if "type" in c and c["type"] in PromptResponseClassification.Type:
                 classifications.append(PromptResponseClassification.from_dict(c))
             else:
                 classifications.append(Classification.from_dict(c))

libs/labelbox/src/labelbox/schema/ontology_kind.py

@@ -1,5 +1,5 @@
 from enum import Enum
-from typing import Optional
+from typing import Optional, Union
 
 from labelbox.schema.media_type import MediaType
 
@@ -24,7 +24,7 @@ def get_ontology_kind_validation_error(cls, ontology_kind):
 
     @staticmethod
     def evaluate_ontology_kind_with_media_type(ontology_kind,
-                                               media_type: MediaType = None) -> MediaType:
+                                               media_type: Union[MediaType, None]) -> MediaType:
 
         if ontology_kind and ontology_kind is OntologyKind.ModelEvaluation:
             if media_type is None:

libs/labelbox/src/labelbox/schema/project.py

@@ -151,7 +151,7 @@ def is_prompt_response(self) -> bool:
         Returns:
             True if this project is a prompt response project, False otherwise
         """
-        return self.media_type in [MediaType.LLMPromptResponseCreation, MediaType.LLMPromptCreation] or EditorTaskType.ResponseCreation
+        return self.media_type == MediaType.LLMPromptResponseCreation or self.media_type == MediaType.LLMPromptCreation or self.editor_task_type == EditorTaskType.ResponseCreation
 
     def is_auto_data_generation(self) -> bool:
         return (self.upload_type == UploadType.Auto)  # type: ignore