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

@@ -32,6 +32,9 @@ class Drift:
}
}
model_config = input_data['model_config']
target_name = model_config['target']
gathering_query = f"""
SELECT *
FROM {input_data['schema']}.{input_data['source_table_name']}
@@ -76,7 +79,7 @@ class Drift:
'reference_data': reference_data,
'model_name': input_data['model_name'],
'model_id': input_data['model_id'],
'target_name': input_data['target_name'],
'target_name': target_name,
'drift_metrics': input_data['drift_metrics'],
'chunk_period': input_data.get('chunk_period', 'min'),
},