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,
         }