lmdb

Genentech · Nov 5, 2024 · 39220ff · 39220ff
1 parent 331870c
commit 39220ff
Showing 1 changed file with 11 additions and 11 deletions.
diff --git a/src/beignet/datasets/_lmdb_dataset.py b/src/beignet/datasets/_lmdb_dataset.py
@@ -8,17 +8,6 @@
 
 from beignet.transforms import Transform
 
-try:
-    import lmdb
-except ImportError as error:
-    raise ImportError(
-        """
-        LMDB datasets require the `lmdb` dependency:
-
-            $ pip install "beignet[lmdb]"
-        """
-    ) from error
-
 
 class LMDBDataset(Dataset):
     def __init__(
@@ -34,6 +23,17 @@ def __init__(
     ):
         super().__init__()
 
+        try:
+            import lmdb
+        except ImportError as error:
+            raise ImportError(
+                """
+                LMDB datasets require the `lmdb` dependency:
+
+                    $ pip install "beignet[lmdb]"
+                """
+            ) from error
+
         self._root = root
 
         self._transform_fn = transform