diff --git a/src/backend/app/services/task/finetune_handler.py b/src/backend/app/services/task/finetune_handler.py index 2c3d26b..060e14b 100644 --- a/src/backend/app/services/task/finetune_handler.py +++ b/src/backend/app/services/task/finetune_handler.py @@ -75,7 +75,7 @@ class FinetuneTaskHandler(BaseTaskHandler): 'class_dir': pm.get_class_data_path(user_id, flow_id), 'coords_save_path': pm.get_uploaded_coords_path(user_id, flow_id, task_id), 'validation_output_dir': pm.get_uploaded_generated_path(user_id, flow_id, task_id), - 'is_perturbed': False, + 'finetune_type': 'uploaded', 'custom_params': None, } @@ -97,7 +97,7 @@ class FinetuneTaskHandler(BaseTaskHandler): 'class_dir': pm.get_class_data_path(user_id, flow_id), 'coords_save_path': pm.get_original_coords_path(user_id, flow_id, task_id), 'validation_output_dir': pm.get_original_generated_path(user_id, flow_id, task_id), - 'is_perturbed': False, + 'finetune_type': 'original', 'custom_params': None, } @@ -110,7 +110,7 @@ class FinetuneTaskHandler(BaseTaskHandler): 'class_dir': pm.get_class_data_path(user_id, flow_id), 'coords_save_path': pm.get_perturbed_coords_path(user_id, flow_id, task_id), 'validation_output_dir': pm.get_perturbed_generated_path(user_id, flow_id, task_id), - 'is_perturbed': True, + 'finetune_type': 'perturbed', 'custom_params': None, }