fix: wrong argilla module import

Author: frascuchon

Diff for: src/distilabel/steps/argilla/base.py

@@ -92,7 +92,7 @@ def model_post_init(self, __context: Any) -> None:
         super().model_post_init(__context)
 
         try:
-            import argilla_sdk as rg  # noqa
+            import argilla as rg  # noqa
         except ImportError as ie:
             raise ImportError(
                 "Argilla is not installed. Please install it using `pip install argilla --upgrade`."

Diff for: tests/unit/steps/argilla/test_preference.py

@@ -15,7 +15,7 @@
 import os
 from unittest.mock import patch
 
-import argilla_sdk as rg
+import argilla as rg
 from distilabel.pipeline.local import Pipeline
 from distilabel.steps.argilla.preference import PreferenceToArgilla
 

Diff for: tests/unit/steps/argilla/test_text_generation.py

@@ -15,7 +15,7 @@
 import os
 from unittest.mock import patch
 
-import argilla_sdk as rg
+import argilla as rg
 from distilabel.pipeline.local import Pipeline
 from distilabel.steps.argilla.text_generation import TextGenerationToArgilla