fix postgres ds get_unsafe return type

opendatacube · Oct 23, 2024 · 0de2e24 · 0de2e24
1 parent 0650c4c
commit 0de2e24
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/datacube/index/postgres/_datasets.py b/datacube/index/postgres/_datasets.py
@@ -65,7 +65,9 @@ def get_unsafe(self, id_: DSID, include_sources: bool = False,
         with self._db_connection() as connection:
             if not include_sources:
                 dataset = connection.get_dataset(id_)
-                return self._make(dataset, full_info=True) if dataset else None
+                if not dataset:
+                    raise KeyError(id_)
+                return self._make(dataset, full_info=True)
 
             datasets = {result.id: (self._make(result, full_info=True), result)
                         for result in connection.get_dataset_sources(id_)}