Fix typo in readme and remove the ToArgilla step (#548)

* Fix typo in readme and remove the ToArgilla step * Update README.md * Update index.md * Update README.md
argilla-io · Apr 17, 2024 · ffc0f4f · ffc0f4f
1 parent 7bbc8ee
commit ffc0f4f
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 32 deletions.
diff --git a/README.md b/README.md
@@ -91,18 +91,18 @@ In addition, the following extras are available:
 
 ### Example
 
-To run the following example you must install `distilabel` with both `openai` and `argilla` extras:
+To run the following example you must install `distilabel` with both `openai` extra:
 
 ```sh
-pip install "distilabel[openai,argilla]" --upgrade
+pip install "distilabel[openai]" --upgrade
 ```
 
-Then run the following example:
+Then run:
 
 ```python
 from distilabel.llms import OpenAILLM
 from distilabel.pipeline import Pipeline
-from distilabel.steps import LoadHubDataset, TextGenerationToArgilla
+from distilabel.steps import LoadHubDataset
 from distilabel.steps.tasks import TextGeneration
 
 with Pipeline(
@@ -115,14 +115,10 @@ with Pipeline(
     )
 
     generate_with_openai = TextGeneration(
-        name="generate_with_gpt3.5", llm=OpenAILLM(model="gpt-3.5-turbo")
+        name="generate_with_gpt35", llm=OpenAILLM(model="gpt-3.5-turbo")
     )
 
-    to_argilla = TextGenerationToArgilla(name="to_argilla")
-
     load_dataset.connect(generate_with_openai)
-    generate_with_openai.connect(to_argilla)
-
 
 if __name__ == "__main__":
     distiset = pipeline.run(
@@ -139,20 +135,10 @@ if __name__ == "__main__":
                     }
                 }
             },
-            "to_argilla": {
-                "dataset_name": "text-generations-with-gpt35",
-                "dataset_workspace": "admin",
-            },
         },
     )
-    distiset.push_to_hub(
-        "distilabel-internal-testing/instruction-dataset-mini-with-generations"
-    )
 ```
 
-Distilabel integrates smoothly with Argilla and provides all the necessary configurations to make giving a final human touch as easy as possible.
-
-
 ## Badges
 
 If you build something cool with `distilabel` consider adding one of these badges to your dataset or model card.

diff --git a/docs/index.md b/docs/index.md
@@ -34,10 +34,18 @@ In addition, the following extras are available:
 
 ## Quick example
 
+To run the following example you must install `distilabel` with both `openai` extra:
+
+```sh
+pip install "distilabel[openai]" --upgrade
+```
+
+Then run:
+
 ```python
 from distilabel.llms import OpenAILLM
 from distilabel.pipeline import Pipeline
-from distilabel.steps import LoadHubDataset, TextGenerationToArgilla
+from distilabel.steps import LoadHubDataset
 from distilabel.steps.tasks import TextGeneration
 
 with Pipeline(
@@ -50,14 +58,10 @@ with Pipeline(
     )
 
     generate_with_openai = TextGeneration(
-        name="generate_with_gpt3.5", llm=OpenAILLM(model="gpt-3.5-turbo")
+        name="generate_with_gpt35", llm=OpenAILLM(model="gpt-3.5-turbo")
     )
 
-    to_argilla = TextGenerationToArgilla(name="to_argilla")
-
     load_dataset.connect(generate_with_openai)
-    generate_with_openai.connect(to_argilla)
-
 
 if __name__ == "__main__":
     distiset = pipeline.run(
@@ -74,13 +78,6 @@ if __name__ == "__main__":
                     }
                 }
             },
-            "to_argilla": {
-                "dataset_name": "text-generations-with-gpt35",
-                "dataset_workspace": "admin",
-            },
         },
     )
-    distiset.push_to_hub(
-        "distilabel-internal-testing/instruction-dataset-mini-with-generations"
-    )
 ```