Fix TaskAnnotationState and Optimization JobType

geti_sdk/data_models/enums/job_type.py

@@ -26,6 +26,8 @@ class JobType(Enum):
     RECONSTRUCT_VIDEO = "reconstruct_video"
     EVALUATE = "evaluate"
     OPTIMIZATION = "optimization"
+    OPTIMIZE_POT = "optimize_pot"
+    OPTIMIZE_NNCF = "optimize_nncf"
     TEST = "test"
     PREPARE_IMPORT_TO_NEW_PROJECT = "prepare_import_to_new_project"
     PERFORM_IMPORT_TO_NEW_PROJECT = "perform_import_to_new_project"

geti_sdk/data_models/task_annotation_state.py

@@ -11,6 +11,7 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions
 # and limitations under the License.
+from typing import Optional
 
 import attr
 
@@ -26,6 +27,9 @@ class TaskAnnotationState:
     """
 
     task_id: str
-    state: str = attr.field(
-        converter=str_to_enum_converter_by_name_or_value(AnnotationState)
+    state: Optional[str] = attr.field(
+        converter=str_to_enum_converter_by_name_or_value(
+            AnnotationState, allow_none=True
+        ),
+        default=None,
     )

geti_sdk/data_models/utils.py

@@ -104,7 +104,7 @@ def _converter(input_value: Optional[Union[str, EnumType]]) -> Optional[EnumType
 
 
 def str_to_enum_converter_by_name_or_value(
-    enum: Type[EnumType],
+    enum: Type[EnumType], allow_none: bool = False
 ) -> Callable[[Union[str, EnumType]], EnumType]:
     """
     Construct a converter function to convert an input value into an instance of the
@@ -113,6 +113,7 @@ def str_to_enum_converter_by_name_or_value(
     This method attempts to convert both from the Enum value as well as it's name
 
     :param enum: type of the Enum to which the converter should convert
+    :param allow_none: True to allow `None` as input value, i.e. for optional parameters
     :return: Converter function that takes an input value and attempts to convert it
         into an instance of `enum`
     """
@@ -131,6 +132,8 @@ def _converter(input_value: Union[str, EnumType]) -> EnumType:
                 return enum[input_value]
         elif isinstance(input_value, enum):
             return input_value
+        if (input_value is None) and allow_none:
+            return None
         else:
             raise ValueError(
                 f"Invalid argument! Cannot convert value {input_value} to Enum "