Bump sumo wrapper. Download blobs async

equinor · Oct 23, 2023 · c6305a6 · c6305a6
1 parent 576fb85
commit c6305a6
Show file tree

Hide file tree

Showing 4 changed files with 7 additions and 7 deletions.
diff --git a/backend/poetry.lock b/backend/poetry.lock
diff --git a/backend/pyproject.toml b/backend/pyproject.toml
@@ -24,7 +24,7 @@ httpx = "^0.24.0"
 psutil = "^5.9.5"
 vtk = "^9.2.6"
 fmu-sumo = "^0.5.4"
-sumo-wrapper-python = "^0.4.0"
+sumo-wrapper-python = "^0.4.1"
 
 
 [tool.poetry.group.dev.dependencies]

diff --git a/backend/src/services/sumo_access/parameter_access.py b/backend/src/services/sumo_access/parameter_access.py
@@ -36,7 +36,7 @@ async def get_parameters_and_sensitivities(self) -> EnsembleParameters:
             raise ValueError(f"Multiple parameter tables found {self._case.name,self._iteration_name}")
 
         table = await table_collection.getitem_async(0)
-        byte_stream: BytesIO = table.blob
+        byte_stream: BytesIO = await table.blob_async
         table = pq.read_table(byte_stream)
 
         et_download_arrow_table_ms = timer.lap_ms()

diff --git a/backend/src/services/sumo_access/summary_access.py b/backend/src/services/sumo_access/summary_access.py
@@ -237,7 +237,7 @@ async def _load_arrow_table_for_from_sumo(case: Case, iteration_name: str, vecto
     # The tables we have seen so far have format set to 'arrow', but the actual data is in parquet format.
     # This must be a bug or a misunderstanding.
     # For now, just read the parquet data into an arrow table
-    byte_stream: BytesIO = sumo_table.blob
+    byte_stream: BytesIO = await sumo_table.blob_async
     table = pq.read_table(byte_stream)
     et_download_arrow_table_ms = timer.lap_ms()