Skip to content

Commit 67bcfe1

Browse files
committed
fix: wrong argilla module import
1 parent c9fc2a5 commit 67bcfe1

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

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

+1-1
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@
1515
import os
1616
from unittest.mock import patch
1717

18-
import argilla_sdk as rg
18+
import argilla as rg
1919
from distilabel.pipeline.local import Pipeline
2020
from distilabel.steps.argilla.preference import PreferenceToArgilla
2121

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

+1-1
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@
1515
import os
1616
from unittest.mock import patch
1717

18-
import argilla_sdk as rg
18+
import argilla as rg
1919
from distilabel.pipeline.local import Pipeline
2020
from distilabel.steps.argilla.text_generation import TextGenerationToArgilla
2121

0 commit comments

Comments
 (0)