Code quality fix

Azure · Feb 3, 2025 · d9aef5c · d9aef5c
1 parent b95453e
commit d9aef5c
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 3 deletions.
diff --git a/assets/training/distillation/src/common/student_models.py b/assets/training/distillation/src/common/student_models.py
@@ -85,7 +85,7 @@ def no_system_prompt_reformat_conversation(cls, data: List[Dict[str, list]]):
                         {"role": "user", "content": system_message + " " + user_prompt},
                         messages["messages"][2:]
                     ]
-            }  
+            }
             new_data.append(reformatted_data)
         return new_data
 

diff --git a/assets/training/distillation/src/common/utils.py b/assets/training/distillation/src/common/utils.py
@@ -500,4 +500,4 @@ def get_hash_value(data: Union[Dict[str, Any], str]) -> str:
     """
     if isinstance(data, str):
         return hashlib.sha256(data.encode()).hexdigest()
-    return hashlib.sha256(json.dumps(data).encode()).hexdigest()
+    return hashlib.sha256(json.dumps(data).encode()).hexdigest()
diff --git a/assets/training/distillation/src/generate_data_postprocess.py b/assets/training/distillation/src/generate_data_postprocess.py
@@ -300,7 +300,11 @@ def postprocess_data(
         raise Exception(msg)
 
     # Reformat data based on student model limitations
-    # output_data = StudentModels.reformat(student_model=student_model, task_type=data_generation_task_type, data=output_data)
+    # output_data = StudentModels.reformat(
+    #   student_model=student_model,
+    #   task_type=data_generation_task_type,
+    #   data=output_data
+    # )
     with open(output_file_path, "w") as f:
         for record in output_data:
             f.write(json.dumps(record) + "\n")