additional guard on datasets import

huggingface · Feb 1, 2024 · 3c4989d · 3c4989d
1 parent 70c9305
commit 3c4989d
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/optimum/intel/openvino/quantization.py b/optimum/intel/openvino/quantization.py
@@ -52,7 +52,10 @@
 
 
 if is_datasets_available():
-    from datasets import Dataset
+    try:
+        from datasets import Dataset
+    except ImportError:
+        pass
 
 COMPRESSION_OPTIONS = {
     "int8": {"mode": nncf.CompressWeightsMode.INT8},