convert to using logging.warning

python/morpheus/morpheus/config.py

@@ -241,7 +241,7 @@ def model_max_batch_size(self, value: int):
 
     def _validate_config(self):
         if self._pipeline_batch_size < self._model_max_batch_size:
-            warnings.warn(
+            logging.warning(
                 "Config has `pipeline_batch_size < model_max_batch_size` which effectively limits `model_max_batch_size`. This may reduce performance."
             )
 

tests/test_config.py

@@ -17,6 +17,7 @@
 import json
 import os
 from unittest import mock
+import logging
 
 import pytest
 
@@ -108,15 +109,21 @@ def test_to_string(config):
     assert isinstance(json.loads(conf_str), dict)
 
 
-def test_warning_model_batch_size_less_than_pipeline_batch_size():
+def test_warning_model_batch_size_less_than_pipeline_batch_size(caplog: pytest.LogCaptureFixture):
     config = morpheus.config.Config()
     config.pipeline_batch_size = 256
-    with pytest.warns():
+    with caplog.at_level(logging.WARNING):
         config.model_max_batch_size = 257
+        assert len(caplog.records) == 1
+        import re
+        assert re.match(".*pipeline_batch_size < model_max_batch_size.*", caplog.records[0].message) is not None
 
 
-def test_warning_pipeline_batch_size_less_than_model_batch_size():
+def test_warning_pipeline_batch_size_less_than_model_batch_size(caplog: pytest.LogCaptureFixture):
     config = morpheus.config.Config()
     config.model_max_batch_size = 8
-    with pytest.warns():
+    with caplog.at_level(logging.WARNING):
         config.pipeline_batch_size = 7
+        assert len(caplog.records) == 1
+        import re
+        assert re.match(".*pipeline_batch_size < model_max_batch_size.*", caplog.records[0].message) is not None