From 67bcfe1f3260e986bb68d3613f9b9277f177def2 Mon Sep 17 00:00:00 2001 From: Francisco Aranda Date: Wed, 17 Jul 2024 09:49:06 +0200 Subject: [PATCH] fix: wrong argilla module import --- tests/unit/steps/argilla/test_preference.py | 2 +- tests/unit/steps/argilla/test_text_generation.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/unit/steps/argilla/test_preference.py b/tests/unit/steps/argilla/test_preference.py index 5ff95782b0..edeb566849 100644 --- 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 index 5679c54517..884ae51cff 100644 --- 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