gitea_admin/sientia-dataops-orchestrator_temporal
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

SIENTIAPDE-1182

Browse Source 
SIENTIAPDE-1205: Enhance predictions_batch function and update test cases

- Added 'predictions_storage_policy' to the predictions_batch function for improved configuration handling.
- Updated test cases to include 'predictions_storage_policy' in expected results, ensuring consistency in testing.
This commit is contained in:
vitor-aignosi
2025-08-28 16:32:06 -03:00
parent d701943586
commit 36f1388301
2 changed files with 6 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
orchestrator/utils/orchestrator_functions.py
View File

@@ -211,7 +211,8 @@ def predictions_batch(config: dict[str, Any]):
"config": {} "config": {}
} }
}, config.get('mlflow_predict_filters', [])), }, config.get('mlflow_predict_filters', [])),
"path_priority": path_priority "path_priority": path_priority,
"predictions_storage_policy": config.get('predictions_storage_policy', 'lts:1')
} }

6
tests/orchestrator/utils/test_orchestrator_functions.py
View File

@@ -197,7 +197,8 @@ def test_predictions_batch(mock_process_path_priority,
} }
], ],
"path_priority": ["STOP", "CONTINUE", "REPEAT"], "path_priority": ["STOP", "CONTINUE", "REPEAT"],
"datetime_columns": ["timestamp"] "datetime_columns": ["timestamp"],
"predictions_storage_policy": "erl:1"
} }
result = predictions_batch(config) result = predictions_batch(config)
@@ -272,7 +273,8 @@ def test_predictions_batch(mock_process_path_priority,
} }
}, },
"path_priority": ["STOP", "CONTINUE", "REPEAT"], "path_priority": ["STOP", "CONTINUE", "REPEAT"],
"datetime_columns": ["timestamp"] "datetime_columns": ["timestamp"],
"predictions_storage_policy": "erl:1"
} }
assert result == expected assert result == expected