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