Fix ImportError for argilla in TextGenerationToArgilla

argilla-io · Apr 11, 2024 · 6cbcced · 6cbcced
1 parent 4361fbd
commit 6cbcced
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/src/distilabel/steps/argilla/text_generation.py b/src/distilabel/steps/argilla/text_generation.py
@@ -20,10 +20,8 @@
 
 try:
     import argilla as rg
-except ImportError as ie:
-    raise ImportError(
-        "Argilla is not installed. Please install it using `pip install argilla`."
-    ) from ie
+except ImportError:
+    pass
 
 from distilabel.steps.argilla.base import Argilla
 from distilabel.steps.base import StepInput