add logging of when we enter and exit gj

ecmwf · Nov 7, 2024 · bb4e17e · bb4e17e
1 parent 09f6ede
commit bb4e17e
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/polytope_feature/datacube/backends/fdb.py b/polytope_feature/datacube/backends/fdb.py
@@ -29,7 +29,9 @@ def __init__(
 
         self.gj = gj
         if len(alternative_axes) == 0:
+            logging.info("Find GribJump axes for %s", context)
             self.fdb_coordinates = self.gj.axes(partial_request, ctx=context)
+            logging.info("Retrieved available GribJump axes for %s", context)
             if len(self.fdb_coordinates) == 0:
                 raise BadRequestError(partial_request)
         else:
@@ -117,7 +119,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)
+        logging.info("Requests given to GribJump extract for %s", context)
         output_values = self.gj.extract(complete_list_complete_uncompressed_requests, context)
+        logging.info("Requests extracted from GribJump for %s", context)
         logging.debug("GribJump outputs: %s", output_values)
         self.assign_fdb_output_to_nodes(output_values, complete_fdb_decoding_info)