fix blob download and handle exception in compute (#12152)

GitOrigin-RevId: 27b7eb6365b1a5d08678d86c147d6e797754a592

descarteslabs/core/catalog/blob.py

@@ -16,6 +16,8 @@
 
 from strenum import StrEnum
 
+from descarteslabs.exceptions import NotFoundError
+
 from ..client.services.service import ThirdPartyService
 from ..common.collection import Collection
 from ..common.property_filtering import Properties
@@ -30,14 +32,9 @@
     parse_iso_datetime,
 )
 from .blob_download import BlobDownload
-from .catalog_base import (
-    CatalogClient,
-    CatalogObject,
-    check_deleted,
-)
+from .catalog_base import CatalogClient, CatalogObject, check_deleted
 from .search import AggregateDateField, GeoSearch, SummarySearchMixin
 
-
 properties = Properties()
 
 
@@ -835,6 +832,11 @@ def get_data(
     def _do_download(self, dest=None, range=None):
         download = BlobDownload.get(id=self.id, client=self._client)
 
+        # BlobDownload.get() returns None if the blob does not exist
+        # raise a NotFoundError in this case
+        if not download:
+            raise NotFoundError("Blob {} does not exist".format(self.id))
+
         headers = {}
         if self.hash:
             headers["if-match"] = self.hash

descarteslabs/core/compute/job.py

@@ -19,6 +19,9 @@
 from typing import TYPE_CHECKING, Dict, List, Optional, Type
 
 from strenum import StrEnum
+
+from descarteslabs.exceptions import NotFoundError
+
 from ..catalog import Blob, CatalogClient, DeletedObjectError, StorageType
 from ..common.client import (
     Attribute,
@@ -277,6 +280,8 @@ def result(self, cast_type: Optional[Type[Serializable]] = None):
                 storage_type=StorageType.COMPUTE,
                 client=client,
             )
+        except NotFoundError:
+            return None
         except ValueError:
             raise
         except DeletedObjectError: