SIENTIAPDE-1241: refactor train_model workflow due to I/O errors.

2025-10-22 15:37:56 -03:00
parent f2a1c88ff3
commit 5789a13023
31 changed files with 37878 additions and 6097 deletions
--- a/tests/sientia/test_model_serving.py
+++ b/tests/sientia/test_model_serving.py
@@ -16,9 +16,8 @@ def test_init_with_all_credentials(mock_set_tracking_uri):
    tracking_uri = 'http://mlflow.example.com'
    username = 'test_user'
    password = 'test_pass'
-    logger = MagicMock()

-    ModelServing(tracking_uri=tracking_uri, username=username, password=password, logger=logger)
+    ModelServing(tracking_uri=tracking_uri, username=username, password=password)

    mock_set_tracking_uri.assert_called_once_with(tracking_uri)
    import os