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

SIENTIAPDE-1030

Browse Source 
Refactor path priority handling in predictions_batch function
This commit is contained in:
vitor-aignosi
2025-05-30 16:26:33 -03:00
parent 4f37b2be02
commit 43e9695849
1 changed files with 15 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
orchestrator/activities/formatters.py
View File

@@ -74,6 +74,19 @@ def overlap_filter_config(base_filter_config: dict[str, Any], config: dict[str,
return base_filter_config
def process_path_priority(path_priority: list[str]):
for priority in path_priority[:]:
if priority not in ["STOP", "CONTINUE", "REPEAT"]:
path_priority.remove(priority)
if len(path_priority) != 3:
for priority in ["STOP", "CONTINUE", "REPEAT"]:
if priority not in path_priority:
path_priority.append(priority)
return path_priority
def predictions_batch(config: dict[str, Any]):
tags = {}
for tag in config['write_tags']:
@@ -92,16 +105,8 @@ def predictions_batch(config: dict[str, Any]):
"data_type": tag.get('data_type', 'float'),
}
path_priority = config.get('path_priority', ["STOP", "CONTINUE", "REPEAT"])
for priority in path_priority[:]:
if priority not in ["STOP", "CONTINUE", "REPEAT"]:
path_priority.remove(priority)
if len(path_priority) != 3:
for priority in ["STOP", "CONTINUE", "REPEAT"]:
if priority not in path_priority:
path_priority.append(priority)
path_priority = process_path_priority(config.get(
'path_priority', ["STOP", "CONTINUE", "REPEAT"]))
return {
**common_config(config),