SIENTIAPDE-1030
Refactor path priority handling in predictions_batch function
This commit is contained in:
@@ -74,6 +74,19 @@ def overlap_filter_config(base_filter_config: dict[str, Any], config: dict[str,
|
|||||||
return base_filter_config
|
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]):
|
def predictions_batch(config: dict[str, Any]):
|
||||||
tags = {}
|
tags = {}
|
||||||
for tag in config['write_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'),
|
"data_type": tag.get('data_type', 'float'),
|
||||||
}
|
}
|
||||||
|
|
||||||
path_priority = config.get('path_priority', ["STOP", "CONTINUE", "REPEAT"])
|
path_priority = process_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)
|
|
||||||
|
|
||||||
return {
|
return {
|
||||||
**common_config(config),
|
**common_config(config),
|
||||||
|
|||||||
Reference in New Issue
Block a user