anemoi-utils 0.3.18__py3-none-any.whl → 0.4.1__py3-none-any.whl

anemoi/utils/__init__.py

@@ -6,4 +6,4 @@
 # nor does it submit to any jurisdiction.
 
 
-from ._version import __version__
+from ._version import __version__ as __version__

anemoi/utils/_version.py

@@ -12,5 +12,5 @@ __version__: str
 __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
 
-__version__ = version = '0.3.18'
-__version_tuple__ = version_tuple = (0, 3, 18)
+__version__ = version = '0.4.1'
+__version_tuple__ = version_tuple = (0, 4, 1)

anemoi/utils/cli.py

@@ -60,6 +60,7 @@ class Failed(Command):
     def __init__(self, name, error):
         self.name = name
         self.error = error
+        traceback.print_tb(error.__traceback__)
 
     def add_arguments(self, command_parser):
         command_parser.add_argument("x", nargs=argparse.REMAINDER)

anemoi/utils/config.py

@@ -353,3 +353,22 @@ def check_config_mode(name="settings.toml", secrets_name=None, secrets=None) ->
             raise SystemError(f"Configuration file {conf} is not secure.\n" f"Please run `chmod 600 {conf}`.")
 
         CHECKED[name] = True
+
+
+def find(metadata, what, result=None):
+    if result is None:
+        result = []
+
+    if isinstance(metadata, list):
+        for i in metadata:
+            find(i, what, result)
+        return result
+
+    if isinstance(metadata, dict):
+        if what in metadata:
+            result.append(metadata[what])
+
+        for k, v in metadata.items():
+            find(v, what, result)
+
+    return result

anemoi/utils/humanize.py

@@ -97,7 +97,7 @@ PERIODS = (
 
 
 def _plural(count):
-    if count > 1:
+    if count != 1:
         return "s"
     else:
         return ""

anemoi/utils/provenance.py

@@ -145,11 +145,15 @@ def _module_versions(full):
 
 
 @cache
-def package_distributions():
+def package_distributions() -> dict[str, list[str]]:
     # Takes a significant amount of time to run
     # so cache the result
     from importlib import metadata
 
+    # For python 3.9 support
+    if not hasattr(metadata, "packages_distributions"):
+        import importlib_metadata as metadata
+
     return metadata.packages_distributions()
 
 

anemoi/utils/text.py

@@ -173,6 +173,12 @@ def green(text):
     return colored(text, "green")
 
 
+def blue(text):
+    from termcolor import colored
+
+    return colored(text, "blue")
+
+
 class Tree:
     """Tree data structure."""
 

{anemoi_utils-0.3.18.dist-info → anemoi_utils-0.4.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: anemoi-utils
-Version: 0.3.18
+Version: 0.4.1
 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
@@ -224,9 +224,11 @@ Classifier: Programming Language :: Python :: Implementation :: PyPy
 Requires-Python: >=3.9
 License-File: LICENSE
 Requires-Dist: aniso8601
+Requires-Dist: python-dateutil
 Requires-Dist: pyyaml
-Requires-Dist: tomli
 Requires-Dist: tqdm
+Requires-Dist: importlib-metadata ; python_version < "3.10"
+Requires-Dist: tomli ; python_version < "3.11"
 Provides-Extra: all
 Requires-Dist: anemoi-utils[grib,provenance,text] ; extra == 'all'
 Provides-Extra: dev

{anemoi_utils-0.3.18.dist-info → anemoi_utils-0.4.1.dist-info}/RECORD

@@ -1,27 +1,27 @@
-anemoi/utils/__init__.py,sha256=zZZpbKIoGWwdCOuo6YSruLR7C0GzvzI1Wzhyqaa0K7M,456
+anemoi/utils/__init__.py,sha256=cXvnZAd--_pmNMnaRQDb7Lqf_tVqdq0QXDC674EKxks,471
 anemoi/utils/__main__.py,sha256=cLA2PidDTOUHaDGzd0_E5iioKYNe-PSTv567Y2fuwQk,723
-anemoi/utils/_version.py,sha256=QAzjzlQ8v3ZqOKXdT0jblmkqebjcDj2MPKGBvc_ei_4,413
+anemoi/utils/_version.py,sha256=H6xAc33bJwmO0oYU-O7_wWInmGAifNGi6CO6khT0G1g,411
 anemoi/utils/caching.py,sha256=bCOjP1jcDbwi7BID3XaR9BbkD1k3UipkP74NbgJuEFA,1974
 anemoi/utils/checkpoints.py,sha256=qWtqkQvMDlPIcfqz_GmOyUf0mmHZ6QEInpvgSJYIQjY,5275
-anemoi/utils/cli.py,sha256=9TrBXkDjBsos2d0z6wdFDRZIjft3HWGutSgAFi0zCK4,3712
-anemoi/utils/config.py,sha256=fHM6kENZhdD350WaEDekSdH_Fs0diovj-nPuv_a7bko,9408
+anemoi/utils/cli.py,sha256=Or4pUnuFqpSTlkTvh_VqV5moP2aBHKYINomPkUJFS68,3760
+anemoi/utils/config.py,sha256=nOYVUmajh-9nlzX-fCCPjVCUDjjaL3zZ8SWr2gP8qNs,9810
 anemoi/utils/dates.py,sha256=dgGbTqpGOpYDGgWfXL_69HutXTCHDFI2DhvG-9I9WQI,12341
 anemoi/utils/grib.py,sha256=mrk1drJm2jaPYERQX45QfX2MP4eUqRv0J-Y8IRSRTRE,3073
 anemoi/utils/hindcasts.py,sha256=X8k-81ltmkTDHdviY0SJgvMg7XDu07xoc5ALlUxyPoo,1453
-anemoi/utils/humanize.py,sha256=-xQraQWMLwNaLQAWfPi4K05qieQLgkiyYmV6bfhr10U,16611
-anemoi/utils/provenance.py,sha256=N0JYslg7foDgkQFGDytkYZhiDMELvJHzGYynetF0rW8,10726
+anemoi/utils/humanize.py,sha256=IxPpplDPSaFFRbAsy84fmit0rxRuJX5XpPhZbBHPB00,16612
+anemoi/utils/provenance.py,sha256=iSq4Y5eoWK_SNBMpMfInOB3shQlMZqE8iMBp-LgKtJk,10882
 anemoi/utils/s3.py,sha256=MuY-PrHpt6iKM2RK7v74YoCdqvVJ8UjBDJh0wxUR9Co,18720
 anemoi/utils/sanitise.py,sha256=VKIUiwm0EHPdkFUR6FkAxe94933yQx2obQtN6YROH5M,2862
 anemoi/utils/sanitize.py,sha256=6HJrfCMnrmH5lfxydcBc-AjxkmkFXdCc1wotzm9NBCw,488
-anemoi/utils/text.py,sha256=WvxkRlPpmK7HLOpWZmyqPQG29GMF8IFaCJp7frfIWNI,10436
+anemoi/utils/text.py,sha256=2D2ZbAdoivCgIjktpLoxb1D62GfyZ8P5KwVS3KWixR8,10522
 anemoi/utils/timer.py,sha256=EQcucuwUaGeSpt2S1APJlwSOu6kC47MK9f4h-r8c_AY,990
 anemoi/utils/commands/__init__.py,sha256=qAybFZPBBQs0dyx7dZ3X5JsLpE90pwrqt1vSV7cqEIw,706
 anemoi/utils/commands/config.py,sha256=KEffXZh0ZQfn8t6LXresfd94kDY0gEyulx9Wto5ttW0,824
 anemoi/utils/mars/__init__.py,sha256=RAeY8gJ7ZvsPlcIvrQ4fy9xVHs3SphTAPw_XJDtNIKo,1750
 anemoi/utils/mars/mars.yaml,sha256=R0dujp75lLA4wCWhPeOQnzJ45WZAYLT8gpx509cBFlc,66
-anemoi_utils-0.3.18.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-anemoi_utils-0.3.18.dist-info/METADATA,sha256=uhfZ-zuzer9SQNNPAvvs-KaqRomfEiJYlXVX1VKdbhM,15055
-anemoi_utils-0.3.18.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
-anemoi_utils-0.3.18.dist-info/entry_points.txt,sha256=LENOkn88xzFQo-V59AKoA_F_cfYQTJYtrNTtf37YgHY,60
-anemoi_utils-0.3.18.dist-info/top_level.txt,sha256=DYn8VPs-fNwr7fNH9XIBqeXIwiYYd2E2k5-dUFFqUz0,7
-anemoi_utils-0.3.18.dist-info/RECORD,,
+anemoi_utils-0.4.1.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+anemoi_utils-0.4.1.dist-info/METADATA,sha256=PoA_pFFYtvbx-ygG4g4UumwRLJ8kQx32xIadEE_6iSg,15171
+anemoi_utils-0.4.1.dist-info/WHEEL,sha256=OVMc5UfuAQiSplgO0_WdW7vXVGAt9Hdd6qtN4HotdyA,91
+anemoi_utils-0.4.1.dist-info/entry_points.txt,sha256=LENOkn88xzFQo-V59AKoA_F_cfYQTJYtrNTtf37YgHY,60
+anemoi_utils-0.4.1.dist-info/top_level.txt,sha256=DYn8VPs-fNwr7fNH9XIBqeXIwiYYd2E2k5-dUFFqUz0,7
+anemoi_utils-0.4.1.dist-info/RECORD,,

{anemoi_utils-0.3.18.dist-info → anemoi_utils-0.4.1.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (75.1.0)
+Generator: setuptools (75.2.0)
 Root-Is-Purelib: true
 Tag: py3-none-any