SIENTIAPDE-1273

SIENTIAPDE-1273 Add 'save_transform' and 'transform_table_name' parameters to predictions_batch function for enhanced data processing capabilities. Update corresponding test cases to reflect these changes.
This commit is contained in:
vitor-aignosi
2025-11-19 08:12:08 -03:00
parent 81732d1409
commit 62e7a7dee9
2 changed files with 4 additions and 0 deletions

View File

@@ -248,6 +248,8 @@ def predictions_batch(config: dict[str, Any]):
'datetime_columns': config.get('datetime_columns', []),
'schema': 'sientia_data',
'table_name': 'predictions',
'save_transform': config.get('save_transform', True),
'transform_table_name': 'transformed_data',
'retention_time': config.get('model_retention_minutes', 60) * 60,
'opc_output_config': tags,
'input_filters': overlap_filter_config(

View File

@@ -238,6 +238,8 @@ def test_predictions_batch(mock_process_path_priority, mock_overlap_filter_confi
'query': 'test_query',
'schema': 'sientia_data',
'table_name': 'predictions',
'save_transform': True,
'transform_table_name': 'transformed_data',
'retention_time': 60 * 60,
'opc_output_config': {
'test_server_id': {