From c9b7f4e690f94a28f73e26b8eb11bdae8c94a46e Mon Sep 17 00:00:00 2001 From: Mert <101130780+mertalev@users.noreply.github.com> Date: Sun, 11 Feb 2024 01:12:11 -0500 Subject: [PATCH] chore(ml): make execution provider log info-level (#7024) * change debug log to info * update test --- machine-learning/app/models/base.py | 2 +- machine-learning/app/test_main.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/machine-learning/app/models/base.py b/machine-learning/app/models/base.py index 1c15a9d41..df9a612e2 100644 --- a/machine-learning/app/models/base.py +++ b/machine-learning/app/models/base.py @@ -217,7 +217,7 @@ class InferenceModel(ABC): @provider_options.setter def provider_options(self, provider_options: list[dict[str, Any]]) -> None: - log.debug(f"Setting execution provider options to {provider_options}") + log.info(f"Setting execution provider options to {provider_options}") self._provider_options = provider_options @property diff --git a/machine-learning/app/test_main.py b/machine-learning/app/test_main.py index adcb5f43c..c43e661f4 100644 --- a/machine-learning/app/test_main.py +++ b/machine-learning/app/test_main.py @@ -170,7 +170,7 @@ class TestBase: encoder.clear_cache() mock_rmtree.assert_called_once_with(encoder.cache_dir) - info.assert_called_once() + assert info.call_count == 2 def test_clear_cache_warns_if_path_does_not_exist(self, mocker: MockerFixture) -> None: mock_rmtree = mocker.patch("app.models.base.rmtree", autospec=True)