adding helpful error when a dataset fails to load

jim-meyer · Aug 25, 2022 · ef536b1 · ef536b1
1 parent 6bedc9c
commit ef536b1
Showing 1 changed file with 20 additions and 0 deletions.
diff --git a/fiftyone/core/dataset.py b/fiftyone/core/dataset.py
@@ -4910,6 +4910,25 @@ def _load_dataset(name, virtual=False):
     if not virtual:
         fomi.migrate_dataset_if_necessary(name)
 
+    try:
+        return _do_load_dataset(name, virtual=virtual)
+    except Exception as e:
+        try:
+            version = fomi.get_dataset_revision(name)
+        except:
+            raise e
+
+        if version != focn.VERSION:
+            raise ValueError(
+                "Failed to load dataset '%s' from v%s using client v%s. "
+                "You may need to upgrade your client"
+                % (name, version, focn.VERSION)
+            ) from e
+
+        raise e
+
+
+def _do_load_dataset(name, virtual=False):
     try:
         # pylint: disable=no-member
         dataset_doc = foo.DatasetDocument.objects.get(name=name)
@@ -4968,6 +4987,7 @@ def _load_dataset(name, virtual=False):
             ]
 
     dataset_doc.save()
+
     return dataset_doc, sample_doc_cls, frame_doc_cls