fix to give context to gj.extract

ecmwf · Sep 18, 2024 · cbd9524 · cbd9524
1 parent 05f1c4c
commit cbd9524
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/polytope/datacube/backends/fdb.py b/polytope/datacube/backends/fdb.py
@@ -110,9 +110,9 @@ def get(self, requests: TensorIndexTree, context=None):
                 complete_list_complete_uncompressed_requests.append(complete_uncompressed_request)
                 complete_fdb_decoding_info.append(fdb_requests_decoding_info[j])
         logging.debug("The requests we give GribJump are: %s", complete_list_complete_uncompressed_requests)
-        output_values = self.gj.extract(complete_list_complete_uncompressed_requests)
+        output_values = self.gj.extract(complete_list_complete_uncompressed_requests, context)
         logging.debug("GribJump outputs: %s", output_values)
-        self.assign_fdb_output_to_nodes(output_values, complete_fdb_decoding_info, context)
+        self.assign_fdb_output_to_nodes(output_values, complete_fdb_decoding_info)
 
     def get_fdb_requests(
         self,