fix import (#1181)

catalyst-team · Apr 18, 2021 · 94024a4 · 94024a4
1 parent 54bfc8f
commit 94024a4
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/catalyst/contrib/datasets/__init__.py b/catalyst/contrib/datasets/__init__.py
@@ -2,10 +2,11 @@
 
 from catalyst.settings import SETTINGS
 
-from catalyst.contrib.datasets.market1501 import (
-    Market1501MLDataset,
-    Market1501QGDataset,
-)
+if SETTINGS.cv_required:  # we need imread function here
+    from catalyst.contrib.datasets.market1501 import (
+        Market1501MLDataset,
+        Market1501QGDataset,
+    )
 
 from catalyst.contrib.datasets.mnist import (
     MnistMLDataset,