Address feedback

voxel51 · Jul 2, 2024 · 21f7a8c · 21f7a8c
1 parent 83e0f52
commit 21f7a8c
Showing 1 changed file with 5 additions and 6 deletions.
diff --git a/fiftyone/core/dataset.py b/fiftyone/core/dataset.py
@@ -148,13 +148,12 @@ def load_dataset(name, create_if_missing=False):
     Returns:
         a :class:`Dataset`
     """
-    try:
+    if dataset_exists:
         return Dataset(name, _create=False)
-    except MissingDatasetError as ex:
-        if create_if_missing:
-            return Dataset(name)
-        else:
-            raise ex
+    elif create_if_missing:
+        return Dataset(name)
+    else:
+        raise MissingDatasetError(name)
 
 
 def get_default_dataset_name():