polytope-python 1.0.10__tar.gz → 1.0.11__tar.gz

{polytope-python-1.0.10/polytope_python.egg-info → polytope-python-1.0.11}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 1.0
 Name: polytope-python
-Version: 1.0.10
+Version: 1.0.11
 Summary: Polytope datacube feature extraction library
 Home-page: https://github.com/ecmwf/polytope
 Author: ECMWF

{polytope-python-1.0.10 → polytope-python-1.0.11}/polytope_feature/datacube/backends/datacube.py

@@ -148,17 +148,19 @@ class Datacube(ABC):
         return path
 
     @staticmethod
-    def create(datacube, config={}, axis_options={}, compressed_axes_options=[], alternative_axes=[]):
+    def create(datacube, config={}, axis_options={}, compressed_axes_options=[], alternative_axes=[], context=None):
         # TODO: get the configs as None for pre-determined value and change them to empty dictionary inside the function
         if type(datacube).__name__ == "DataArray":
             from .xarray import XArrayDatacube
 
-            xadatacube = XArrayDatacube(datacube, axis_options, compressed_axes_options)
+            xadatacube = XArrayDatacube(datacube, axis_options, compressed_axes_options, context)
             return xadatacube
         if type(datacube).__name__ == "GribJump":
             from .fdb import FDBDatacube
 
-            fdbdatacube = FDBDatacube(datacube, config, axis_options, compressed_axes_options, alternative_axes)
+            fdbdatacube = FDBDatacube(
+                datacube, config, axis_options, compressed_axes_options, alternative_axes, context
+            )
             return fdbdatacube
 
     def check_branching_axes(self, request):

{polytope-python-1.0.10 → polytope-python-1.0.11}/polytope_feature/datacube/backends/fdb.py

@@ -3,14 +3,19 @@ import operator
 from copy import deepcopy
 from itertools import product
 
+from ...utility.exceptions import BadRequestError
 from ...utility.geometry import nearest_pt
 from .datacube import Datacube, TensorIndexTree
 
 
 class FDBDatacube(Datacube):
-    def __init__(self, gj, config=None, axis_options=None, compressed_axes_options=[], alternative_axes=[]):
+    def __init__(
+        self, gj, config=None, axis_options=None, compressed_axes_options=[], alternative_axes=[], context=None
+    ):
         if config is None:
             config = {}
+        if context is None:
+            context = {}
 
         super().__init__(axis_options, compressed_axes_options)
 
@@ -24,7 +29,9 @@ class FDBDatacube(Datacube):
 
         self.gj = gj
         if len(alternative_axes) == 0:
-            self.fdb_coordinates = self.gj.axes(partial_request)
+            self.fdb_coordinates = self.gj.axes(partial_request, ctx=context)
+            if len(self.fdb_coordinates) == 0:
+                raise BadRequestError(partial_request)
         else:
             self.fdb_coordinates = {}
             for axis_config in alternative_axes:

{polytope-python-1.0.10 → polytope-python-1.0.11}/polytope_feature/datacube/backends/xarray.py

@@ -9,7 +9,7 @@ from .datacube import Datacube
 class XArrayDatacube(Datacube):
     """Xarray arrays are labelled, axes can be defined as strings or integers (e.g. "time" or 0)."""
 
-    def __init__(self, dataarray: xr.DataArray, axis_options=None, compressed_axes_options=[]):
+    def __init__(self, dataarray: xr.DataArray, axis_options=None, compressed_axes_options=[], context=None):
         super().__init__(axis_options, compressed_axes_options)
         if axis_options is None:
             axis_options = {}

{polytope-python-1.0.10 → polytope-python-1.0.11}/polytope_feature/polytope.py

@@ -39,7 +39,7 @@ class Request:
 
 
 class Polytope:
-    def __init__(self, datacube, engine=None, options=None):
+    def __init__(self, datacube, engine=None, options=None, context=None):
         from .datacube import Datacube
         from .engine import Engine
 
@@ -48,7 +48,11 @@ class Polytope:
 
         axis_options, compressed_axes_options, config, alternative_axes = PolytopeOptions.get_polytope_options(options)
 
-        self.datacube = Datacube.create(datacube, config, axis_options, compressed_axes_options, alternative_axes)
+        self.context = context
+
+        self.datacube = Datacube.create(
+            datacube, config, axis_options, compressed_axes_options, alternative_axes, self.context
+        )
         self.engine = engine if engine is not None else Engine.default()
         self.time = 0
 
@@ -56,14 +60,12 @@ class Polytope:
         """Low-level API which takes a polytope geometry object and uses it to slice the datacube"""
         return self.engine.extract(self.datacube, polytopes)
 
-    def retrieve(self, request: Request, method="standard", context=None):
+    def retrieve(self, request: Request, method="standard"):
         """Higher-level API which takes a request and uses it to slice the datacube"""
-        if context is None:
-            context = {}
-        logging.info("Starting request for %s ", context)
+        logging.info("Starting request for %s ", self.context)
         self.datacube.check_branching_axes(request)
         request_tree = self.engine.extract(self.datacube, request.polytopes())
-        logging.info("Created request tree for %s ", context)
-        self.datacube.get(request_tree, context)
-        logging.info("Retrieved data for %s ", context)
+        logging.info("Created request tree for %s ", self.context)
+        self.datacube.get(request_tree, self.context)
+        logging.info("Retrieved data for %s ", self.context)
         return request_tree

{polytope-python-1.0.10 → polytope-python-1.0.11}/polytope_feature/utility/exceptions.py

@@ -2,6 +2,12 @@ class PolytopeError(Exception):
     pass
 
 
+class BadRequestError(PolytopeError):
+    def __init__(self, pre_path):
+        self.pre_path = pre_path
+        self.message = f"No data for {pre_path} is available on the FDB."
+
+
 class AxisOverdefinedError(PolytopeError, KeyError):
     def __init__(self, axis):
         self.axis = axis

polytope-python-1.0.11/polytope_feature/version.py

@@ -0,0 +1 @@
+__version__ = "1.0.11"

{polytope-python-1.0.10 → polytope-python-1.0.11/polytope_python.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 1.0
 Name: polytope-python
-Version: 1.0.10
+Version: 1.0.11
 Summary: Polytope datacube feature extraction library
 Home-page: https://github.com/ecmwf/polytope
 Author: ECMWF

polytope-python-1.0.10/polytope_feature/version.py

@@ -1 +0,0 @@
-__version__ = "1.0.10"