From 6cbccedfb0952dae65f33f6253de70db403c02c5 Mon Sep 17 00:00:00 2001
From: Alvaro Bartolome <alvaro@argilla.io>
Date: Thu, 11 Apr 2024 10:57:02 +0200
Subject: [PATCH] Fix `ImportError` for `argilla` in `TextGenerationToArgilla`

---
 src/distilabel/steps/argilla/text_generation.py | 6 ++----
 1 file changed, 2 insertions(+), 4 deletions(-)

diff --git a/src/distilabel/steps/argilla/text_generation.py b/src/distilabel/steps/argilla/text_generation.py
index b06df01d85..6e7ca21ec8 100644
--- 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