diff --git a/swh/vault/cookers/base.py b/swh/vault/cookers/base.py index db0b23fa5844328f8a86a5ded61876491154f67d..2c974e8094541920d572aee6ca5477edd4c63b14 100644 --- a/swh/vault/cookers/base.py +++ b/swh/vault/cookers/base.py @@ -12,6 +12,7 @@ from typing import ClassVar, Set from psycopg2.extensions import QueryCanceledError import sentry_sdk +import swh.model.swhids from swh.model.swhids import CoreSWHID, ObjectType from swh.storage.interface import StorageInterface @@ -61,7 +62,7 @@ class BaseVaultCooker(metaclass=abc.ABCMeta): - def cook(): cook the object into a bundle """ - SUPPORTED_OBJECT_TYPES: ClassVar[Set[ObjectType]] + SUPPORTED_OBJECT_TYPES: ClassVar[Set[swh.model.swhids.ObjectType]] BUNDLE_TYPE: ClassVar[str] def __init__(