anemoi-datasets 0.5.5__py3-none-any.whl → 0.5.6__py3-none-any.whl

anemoi/datasets/_version.py

@@ -12,5 +12,5 @@ __version__: str
 __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
 
-__version__ = version = '0.5.5'
-__version_tuple__ = version_tuple = (0, 5, 5)
+__version__ = version = '0.5.6'
+__version_tuple__ = version_tuple = (0, 5, 6)

anemoi/datasets/create/functions/sources/xarray/__init__.py

@@ -42,7 +42,7 @@ def load_one(emoji, context, dates, dataset, options={}, flavour=None, **kwargs)
     We have seen this bug triggered when we run many clients in parallel, for example, when we create a new dataset using `xarray-zarr`.
     """
 
-    context.trace(emoji, dataset, options)
+    context.trace(emoji, dataset, options, kwargs)
 
     if isinstance(dataset, str) and ".zarr" in dataset:
         data = xr.open_zarr(name_to_zarr_store(dataset), **options)

anemoi/datasets/create/functions/sources/xarray/coordinates.py

@@ -244,3 +244,9 @@ class ScalarCoordinate(Coordinate):
     @property
     def mars_names(self):
         return (self.variable.name,)
+
+
+class UnsupportedCoordinate(Coordinate):
+    @property
+    def mars_names(self):
+        return (self.variable.name,)

anemoi/datasets/create/functions/sources/xarray/flavour.py

@@ -18,6 +18,7 @@ from .coordinates import LongitudeCoordinate
 from .coordinates import ScalarCoordinate
 from .coordinates import StepCoordinate
 from .coordinates import TimeCoordinate
+from .coordinates import UnsupportedCoordinate
 from .coordinates import XCoordinate
 from .coordinates import YCoordinate
 from .coordinates import is_scalar
@@ -157,11 +158,13 @@ class CoordinateGuesser:
         if c.shape in ((1,), tuple()):
             return ScalarCoordinate(c)
 
-        raise NotImplementedError(
+        LOG.warning(
             f"Coordinate {coord} not supported\n{axis=}, {name=},"
             f" {long_name=}, {standard_name=}, units\n\n{c}\n\n{type(c.values)} {c.shape}"
         )
 
+        return UnsupportedCoordinate(c)
+
     def grid(self, coordinates, variable):
         lat = [c for c in coordinates if c.is_lat]
         lon = [c for c in coordinates if c.is_lon]

{anemoi_datasets-0.5.5.dist-info → anemoi_datasets-0.5.6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: anemoi-datasets
-Version: 0.5.5
+Version: 0.5.6
 Summary: A package to hold various functions to support training of ML models on ECMWF data.
 Author-email: "European Centre for Medium-Range Weather Forecasts (ECMWF)" <software.support@ecmwf.int>
 License: Apache License

{anemoi_datasets-0.5.5.dist-info → anemoi_datasets-0.5.6.dist-info}/RECORD

@@ -1,6 +1,6 @@
 anemoi/datasets/__init__.py,sha256=Z1gqZWhecLcT0RZQqYBLlz01MUlUZd0kWEj_RavbITM,782
 anemoi/datasets/__main__.py,sha256=cLA2PidDTOUHaDGzd0_E5iioKYNe-PSTv567Y2fuwQk,723
-anemoi/datasets/_version.py,sha256=q2mmmMeZBj38-RlMPxNZk3zblQ8jPjzN_oIC744dsOc,411
+anemoi/datasets/_version.py,sha256=4DWmCh7zMgrkQTSjlCuIoYpBzO1CjYxQmpAxbcgg7vE,411
 anemoi/datasets/fields.py,sha256=95TYAfqS_kZp1bzj84GngurVfyVvziVBy3zpUyoturI,1968
 anemoi/datasets/grids.py,sha256=xgvIbpMGuN2GKi2wIBhOLEMzj940nY9PH-toD0rCmPo,8980
 anemoi/datasets/commands/__init__.py,sha256=qAybFZPBBQs0dyx7dZ3X5JsLpE90pwrqt1vSV7cqEIw,706
@@ -61,11 +61,11 @@ anemoi/datasets/create/functions/sources/tendencies.py,sha256=4hFPGU51KnxXp-KteS
 anemoi/datasets/create/functions/sources/xarray_kerchunk.py,sha256=JsuGjTs5BQdJIPRI8TuUNde680UQkT4GbUhOt6wYy38,1432
 anemoi/datasets/create/functions/sources/xarray_zarr.py,sha256=qI_fEKo28hu_B_qMPx67gigysw8qw5ePMU9P0wbDcIc,531
 anemoi/datasets/create/functions/sources/zenodo.py,sha256=n7_sfZHJVDjVnSwc07muO9kS7TwOM4S8DIlJf5pnDiU,1237
-anemoi/datasets/create/functions/sources/xarray/__init__.py,sha256=Ft3p7k8WEz3kBc26b1hWuk2y_13_SdPuizwKRb6cLi0,2954
-anemoi/datasets/create/functions/sources/xarray/coordinates.py,sha256=0iOu2qKTBfaegLYXQA1EEi2bRDL5m2H1hwYfuLjBQk4,6001
+anemoi/datasets/create/functions/sources/xarray/__init__.py,sha256=0HMGeuI2y-DlQJiikYvBLkZeLwdKCWggAadzXnRiMho,2962
+anemoi/datasets/create/functions/sources/xarray/coordinates.py,sha256=xGLLr-_-rLw_qbtPtITmLFAJaSh-ab7dbNVPa5WJO_8,6121
 anemoi/datasets/create/functions/sources/xarray/field.py,sha256=3xy14ywRxYydvxucDlE8g6PeKfP5mcc5NL-2mGDv9nU,3658
 anemoi/datasets/create/functions/sources/xarray/fieldlist.py,sha256=GnYbierm_TuTPMG9WOn99QBi1DxaFo0hra69KGvSiEg,5589
-anemoi/datasets/create/functions/sources/xarray/flavour.py,sha256=8jkV-wUlze2IVQ2WLnfj9-5INYhat3Gw3ERFOrf_VqM,14667
+anemoi/datasets/create/functions/sources/xarray/flavour.py,sha256=8FHztUfJbbozcwOf1yJmHhzG7dibkCSTd0812zSJM5w,14741
 anemoi/datasets/create/functions/sources/xarray/grid.py,sha256=h3O20PsdpJc9V_oae_WEUw8qMnalV_4yjFNAZHvuxPI,3629
 anemoi/datasets/create/functions/sources/xarray/metadata.py,sha256=TSFQ_2ijVtEiXdwnjYr9wPRsTBwqfo_pNNZDFPXFAts,4558
 anemoi/datasets/create/functions/sources/xarray/time.py,sha256=6IXwu9gI1e4LjZ6L6-Bgnq2LgRXvSmkujFWEFTxOymA,4491
@@ -113,9 +113,9 @@ anemoi/datasets/dates/__init__.py,sha256=-RcoD33OQl4GOL92r7ksT8014KhvURjUCb1C9SN
 anemoi/datasets/dates/groups.py,sha256=7zXEtMwFWFFQarKNyNDLxpTu7OFiZWSYIrvAIeWLEPg,5391
 anemoi/datasets/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 anemoi/datasets/utils/fields.py,sha256=VWf9PPmJ1WU1-oJ5tNkbC3_ik9gqnFxqwrc1EFysBYk,1397
-anemoi_datasets-0.5.5.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-anemoi_datasets-0.5.5.dist-info/METADATA,sha256=IXuERodLJ1TIa4E3qplvDYy2ULOqGzZwvTaIntkLoo8,16324
-anemoi_datasets-0.5.5.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
-anemoi_datasets-0.5.5.dist-info/entry_points.txt,sha256=yR-o-4uiPEA_GLBL81SkMYnUoxq3CAV3hHulQiRtGG0,66
-anemoi_datasets-0.5.5.dist-info/top_level.txt,sha256=DYn8VPs-fNwr7fNH9XIBqeXIwiYYd2E2k5-dUFFqUz0,7
-anemoi_datasets-0.5.5.dist-info/RECORD,,
+anemoi_datasets-0.5.6.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+anemoi_datasets-0.5.6.dist-info/METADATA,sha256=9t0mkhplUrznbX7XR2L4k_oLeM1_nfxTz0l-GT11Jlg,16324
+anemoi_datasets-0.5.6.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
+anemoi_datasets-0.5.6.dist-info/entry_points.txt,sha256=yR-o-4uiPEA_GLBL81SkMYnUoxq3CAV3hHulQiRtGG0,66
+anemoi_datasets-0.5.6.dist-info/top_level.txt,sha256=DYn8VPs-fNwr7fNH9XIBqeXIwiYYd2E2k5-dUFFqUz0,7
+anemoi_datasets-0.5.6.dist-info/RECORD,,