From 8b800b08009720744478a78eebb2d4b5d0f06de6 Mon Sep 17 00:00:00 2001 From: Daniel Date: Wed, 4 Jun 2025 15:19:22 +0200 Subject: [PATCH] Add missing DaShare --- da/api/test_flow.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/da/api/test_flow.py b/da/api/test_flow.py index 14f61ed..a52c883 100644 --- a/da/api/test_flow.py +++ b/da/api/test_flow.py @@ -2,6 +2,7 @@ from unittest import TestCase from collections import defaultdict from da.api.common import * +from da.verifier import DAShare @dataclass @@ -25,7 +26,7 @@ class MockStore(BlobStore): self.app_id_store[metadata.app_id][metadata.index] = cert_id # Implements `get_multiple` method from BlobStore abstract class. - def get_multiple(self, app_id, indexes) -> List[Optional[DABlob]]: + def get_multiple(self, app_id, indexes) -> List[Optional[DAShare]]: return [ self.blob_store.get(self.app_id_store[app_id].get(i), None) if self.app_id_store[app_id].get(i) else None for i in indexes ]