fix typing

immich-app · Feb 3, 2024 · 50fd764 · 50fd764
1 parent 16ba6ca
commit 50fd764
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/machine-learning/app/models/base.py b/machine-learning/app/models/base.py
@@ -182,7 +182,7 @@ def provider_options_default(self) -> list[dict[str, Any]]:
                 case "CPUExecutionProvider" | "CUDAExecutionProvider":
                     option = {"arena_extend_strategy": "kSameAsRequested"}
                 case "OpenVINOExecutionProvider":
-                    option = {"disable_dynamic_shapes": True}
+                    option = {"disable_dynamic_shapes": "True"}
                     try:
                         device_ids: list[str] = ort.capi._pybind_state.get_available_openvino_device_ids()
                         log.debug(f"Available OpenVINO devices: {device_ids}")

diff --git a/machine-learning/app/test_main.py b/machine-learning/app/test_main.py
@@ -68,7 +68,7 @@ def test_sets_default_provider_options(self) -> None:
         encoder = OpenCLIPEncoder("ViT-B-32__openai", providers=["OpenVINOExecutionProvider", "CPUExecutionProvider"])
 
         assert encoder.provider_options == [
-            {"disable_dynamic_shapes": True},
+            {"disable_dynamic_shapes": "True"},
             {"arena_extend_strategy": "kSameAsRequested"},
         ]