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:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user