SIENTIAPDE-1273

Refactor Drift class to improve target name handling

- Extracted target name from model configuration in the Drift class for better clarity and maintainability.
- Updated test cases to reflect changes in input data structure, ensuring consistency in target name retrieval.
This commit is contained in:
vitor-aignosi
2025-11-14 08:47:39 -03:00
parent b68674fe64
commit 66193cea15
2 changed files with 5 additions and 2 deletions

View File

@@ -34,7 +34,7 @@ async def test_run(workflow_mock: AsyncMock, drift: Drift):
'source_table_name': 'test_source_table',
'target_table_name': 'test_target_table',
'interval': 60,
'target_name': 'test_target',
'model_config': {'target': 'test_target'},
'drift_metrics': ['psi', 'ks'],
'chunk_period': 'hour',
}