feat: enhance training and experiment tracking functionality

- Updated `Activities` class to improve garbage collection handling.
- Enhanced error messaging in `ExperimentTracking` for better clarity on update failures.
- Refactored `Training` class to streamline exception handling and improve type hints.
- Introduced new methods in `TrainModelParams` for better handling of experiment run IDs and model metadata.
- Added functionality to extract model equations in `DataManagerRepository` for linear regression models.
This commit is contained in:
vitor-aignosi
2026-04-06 15:05:57 -03:00
parent 1352d1ac8f
commit 6b1df7c3a7
22 changed files with 1751 additions and 2085 deletions

View File

@@ -1,9 +1,11 @@
from os import environ
from unittest.mock import patch
from model_manager.utils.connectors_config import (
build_minio_config,
build_mlflow_config,
build_mongodb_config,
build_plugin_store_config,
build_postgres_config,
)
@@ -141,6 +143,13 @@ def test_build_minio_config_with_env_vars():
assert config['read_timeout'] == 120
def test_build_plugin_store_config_cache_ttl_seconds():
"""STORE_CACHE_TTL_SECONDS is parsed to int when set."""
with patch.dict(environ, {'STORE_CACHE_TTL_SECONDS': '7200'}, clear=False):
cfg = build_plugin_store_config()
assert cfg['cache_ttl_seconds'] == 7200
def test_build_minio_config_with_defaults():
# Arrange
# Clear any existing env vars