Fix unit tests

openvinotoolkit · Apr 17, 2024 · 8ed5146 · 8ed5146
1 parent e1bd05a
commit 8ed5146
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/tests/unit/core/model/test_visual_prompting.py b/tests/unit/core/model/test_visual_prompting.py
@@ -276,10 +276,10 @@ class TestOVVisualPromptingModel:
     def set_ov_visual_prompting_model(self, mocker, tmpdir):
         def ov_visual_prompting_model(for_create_model: bool = False) -> OVVisualPromptingModel:
             if for_create_model:
-                mocker.patch("openvino.model_api.adapters.create_core")
-                mocker.patch("openvino.model_api.adapters.get_user_config")
-                mocker.patch("openvino.model_api.adapters.OpenvinoAdapter")
-                mocker.patch("openvino.model_api.models.Model.create_model")
+                mocker.patch("model_api.adapters.create_core")
+                mocker.patch("model_api.adapters.get_user_config")
+                mocker.patch("model_api.adapters.OpenvinoAdapter")
+                mocker.patch("model_api.models.Model.create_model")
             else:
                 mocker.patch.object(
                     OVVisualPromptingModel,

diff --git a/tests/unit/core/utils/test_tile.py b/tests/unit/core/utils/test_tile.py
@@ -24,7 +24,7 @@ def test_tile_transform_consistency(mocker):
     dm_image = Image.from_numpy(np_image)
 
     mock_model = MagicMock(spec=Model)
-    mocker.patch("openvino.model_api.tilers.tiler.Tiler.__init__", return_value=None)
+    mocker.patch("model_api.tilers.tiler.Tiler.__init__", return_value=None)
     mocker.patch.multiple(Tiler, __abstractmethods__=set())
 
     tiler = Tiler(model=mock_model)