Spaces:
Running
Running
File size: 1,117 Bytes
06d4ee9 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 |
"""
Task name mapping utilities for displaying user-friendly task names.
"""
# Mapping from original task names to display names
task_display_names = {
"Perception Temporal Action Loc": "Temporal Action Localisation",
"Llm Merging": "LLM Merging",
"Meta Learning": "Meta Learning",
"Product Recommendation": "Next Product Recommendation",
"Machine Unlearning": "Machine Unlearning",
"Backdoor Trigger Recovery": "Backdoor Trigger Recovery"
}
def get_display_name(task_name):
"""
Get the display name for a task
Args:
task_name (str): The original task name
Returns:
str: The display name for the task
"""
return task_display_names.get(task_name, task_name)
def get_original_name(display_name):
"""
Get the original task name for a display name
Args:
display_name (str): The display name
Returns:
str: The original task name
"""
# Create a reverse mapping
reverse_mapping = {v: k for k, v in task_display_names.items()}
return reverse_mapping.get(display_name, display_name) |