fix: wrong argilla module import

argilla-io · Jul 17, 2024 · 67bcfe1 · 67bcfe1
1 parent c9fc2a5
commit 67bcfe1
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/tests/unit/steps/argilla/test_preference.py b/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 --git a/tests/unit/steps/argilla/test_text_generation.py b/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