update push dataset

src/distilabel_dataset_generator/apps/sft.py

@@ -71,14 +71,14 @@ def generate_system_prompt(dataset_description, temperature, progress=gr.Progres
 
 
 def generate_sample_dataset(system_prompt, num_turns, progress=gr.Progress()):
-    df = generate_dataset(
+    dataframe = generate_dataset(
         system_prompt=system_prompt,
         num_turns=num_turns,
         num_rows=10,
         progress=progress,
         is_sample=True,
     )
-    return df
+    return dataframe
 
 
 def generate_dataset(
@@ -202,7 +202,7 @@ def push_dataset_to_hub(dataframe, org_name, repo_name, oauth_token, private):
     return original_dataframe
 
 
-def push_dataset_to_argilla(
+def push_dataset(
     org_name: str,
     repo_name: str,
     system_prompt: str,
@@ -400,7 +400,10 @@ with gr.Blocks(css=_LOGGED_OUT_CSS) as app:
                 )
             with gr.Column(scale=3):
                 dataframe = gr.Dataframe(
-                    headers=["prompt", "completion"], wrap=True, height=500, interactive=False
+                    headers=["prompt", "completion"],
+                    wrap=True,
+                    height=500,
+                    interactive=False,
                 )
 
         gr.HTML(value="<hr>")
@@ -445,6 +448,7 @@ with gr.Blocks(css=_LOGGED_OUT_CSS) as app:
                         label="Distilabel Pipeline Code",
                     )
 
+
     load_btn.click(
         fn=generate_system_prompt,
         inputs=[dataset_description, temperature],
@@ -456,7 +460,8 @@ with gr.Blocks(css=_LOGGED_OUT_CSS) as app:
         outputs=[dataframe],
         show_progress=True,
     )
-    
+
+
     btn_apply_to_sample_dataset.click(
         fn=generate_sample_dataset,
         inputs=[system_prompt, num_turns],
@@ -479,7 +484,7 @@ with gr.Blocks(css=_LOGGED_OUT_CSS) as app:
         outputs=[success_message],
         show_progress=True,
     ).success(
-        fn=push_dataset_to_argilla,
+        fn=push_dataset,
         inputs=[
             org_name,
             repo_name,
@@ -500,5 +505,6 @@ with gr.Blocks(css=_LOGGED_OUT_CSS) as app:
         outputs=[pipeline_code_ui],
     )
 
+
     app.load(fn=swap_visibility, outputs=main_ui)
     app.load(fn=get_org_dropdown, outputs=[org_name])

src/distilabel_dataset_generator/apps/textcat.py

@@ -1,5 +1,4 @@
 import json
-import re
 import uuid
 from typing import List, Union
 
@@ -189,7 +188,7 @@ def push_dataset_to_hub(
     )
 
 
-def push_dataset_to_argilla(
+def push_dataset(
     org_name: str,
     repo_name: str,
     system_prompt: str,
@@ -509,7 +508,7 @@ with gr.Blocks(css=_LOGGED_OUT_CSS) as app:
         outputs=[success_message],
         show_progress=True,
     ).success(
-        fn=push_dataset_to_argilla,
+        fn=push_dataset,
         inputs=[
             org_name,
             repo_name,