fix type for weight_file

openvinotoolkit · Mar 22, 2022 · 4129bc4 · 4129bc4
1 parent 4f55cae
commit 4129bc4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/external/mmdetection/detection_tasks/apis/detection/openvino_task.py b/external/mmdetection/detection_tasks/apis/detection/openvino_task.py
@@ -135,7 +135,7 @@ def __init__(
         hparams: OTEDetectionConfig,
         label_schema: LabelSchemaEntity,
         model_file: Union[str, bytes],
-        weight_file: Optional[str, bytes] = None,
+        weight_file: Union[str, bytes, None] = None,
         device: str = "CPU",
         num_requests: int = 1,
     ):