anemoi-utils 0.4.14__py3-none-any.whl → 0.4.16__py3-none-any.whl

anemoi/utils/_version.py

@@ -17,5 +17,5 @@ __version__: str
 __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
 
-__version__ = version = '0.4.14'
-__version_tuple__ = version_tuple = (0, 4, 14)
+__version__ = version = '0.4.16'
+__version_tuple__ = version_tuple = (0, 4, 16)

anemoi/utils/devtools.py

@@ -81,13 +81,11 @@ def plot_values(
     ax.add_feature(cfeature.BORDERS, linestyle=":")
 
     missing_values = np.isnan(values)
-
     if missing_value is None:
-        values = values[~missing_values]
-        longitudes = longitudes[~missing_values]
-        latitudes = latitudes[~missing_values]
-    else:
-        values = np.where(missing_values, missing_value, values)
+        min = np.nanmin(values)
+        missing_value = min - np.abs(min) * 0.001
+
+    values = np.where(missing_values, missing_value, values)
 
     if max_value is not None:
         values = np.where(values > max_value, max_value, values)

anemoi/utils/humanize.py

@@ -48,7 +48,7 @@ def bytes_to_human(n: float) -> str:
     """
     if n < 0:
         sign = "-"
-        n -= 0
+        n = -n
     else:
         sign = ""
 
@@ -411,15 +411,9 @@ def when(
         if years > 1:
             return _("%d years" % (years,))
 
-        month = then.month
-        if now.year != then.year:
-            month -= 12
-
-        d = abs(now.month - month)
-        if d >= 12:
-            return _("a year")
-        else:
-            return _("%d month%s" % (d, _plural(d)))
+        delta = abs(now - then)
+        if delta.days > 1 and delta.days < 30:
+            return _("%d days" % (delta.days,))
 
     return "on %s %d %s %d" % (
         DOW[then.weekday()],

anemoi/utils/registry.py

@@ -88,14 +88,17 @@ class Registry:
         The package name.
     key : str, optional
         The key to use for the registry, by default "_type".
+    api_version : str, optional
+        The API version, by default '1.0.0'.
     """
 
-    def __init__(self, package: str, key: str = "_type"):
+    def __init__(self, package: str, key: str = "_type", api_version: str = "1.0.0"):
         self.package = package
         self.__registered = {}
         self._sources = {}
         self.kind = package.split(".")[-1]
         self.key = key
+        self.api_version = api_version
         _BY_KIND[self.kind] = self
 
     @classmethod
@@ -133,6 +136,9 @@ class Registry:
         Wrapper, optional
             A wrapper if the factory is None, otherwise None.
         """
+
+        name = name.replace("_", "-")
+
         if factory is None:
             # This happens when the @register decorator is used
             return Wrapper(name, self)
@@ -162,7 +168,7 @@ class Registry:
         except Exception as e:
             if DEBUG_ANEMOI_REGISTRY:
                 raise
-            self._registered[name] = Error(e)
+            self.__registered[name] = Error(e)
 
     def is_registered(self, name: str) -> bool:
         """Check if a factory is registered.
@@ -177,6 +183,9 @@ class Registry:
         bool
             Whether the factory is registered.
         """
+
+        name = name.replace("_", "-")
+
         ok = name in self.factories
         if not ok:
             LOG.error(f"Cannot find '{name}' in {self.package}")
@@ -199,6 +208,9 @@ class Registry:
         Callable, optional
             The factory if found, otherwise None.
         """
+
+        name = name.replace("_", "-")
+
         if return_none:
             return self.factories.get(name)
 
@@ -288,6 +300,9 @@ class Registry:
         Any
             The created instance.
         """
+
+        name = name.replace("_", "-")
+
         factory = self.lookup(name)
         return factory(*args, **kwargs)
 

{anemoi_utils-0.4.14.dist-info → anemoi_utils-0.4.16.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: anemoi-utils
-Version: 0.4.14
+Version: 0.4.16
 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_utils-0.4.14.dist-info → anemoi_utils-0.4.16.dist-info}/RECORD

@@ -1,20 +1,20 @@
 anemoi/utils/__init__.py,sha256=uVhpF-VjIl_4mMywOVtgTutgsdIsqz-xdkwxeMhzuag,730
 anemoi/utils/__main__.py,sha256=6LlE4MYrPvqqrykxXh7XMi50UZteUY59NeM8P9Zs2dU,910
-anemoi/utils/_version.py,sha256=ggKLQlA5gegh3l4LAMWuwHPmL8zgZ4b0pYtFx7quk78,513
+anemoi/utils/_version.py,sha256=aZNmGWWE7ndWaFJGD1G8axbqrCNEX3UMJLUyVC4HPEM,513
 anemoi/utils/caching.py,sha256=rXbeAmpBcMbbfN4EVblaHWKicsrtx1otER84FEBtz98,6183
 anemoi/utils/checkpoints.py,sha256=N4WpAZXa4etrpSEKhHqUUtG2-x9w3FJMHcLO-dDAXPY,9600
 anemoi/utils/cli.py,sha256=IyZfnSw0u0yYnrjOrzvm2RuuKvDk4cVb8pf8BkaChgA,6209
 anemoi/utils/compatibility.py,sha256=wRBRMmxQP88rNcWiP5gqXliwYQbBv1iCAsDjcCRi5UY,2234
 anemoi/utils/config.py,sha256=4K4_ePeuUPyIMsE7-r77OC87ntFGdVLskB0ow9XNzXI,16353
 anemoi/utils/dates.py,sha256=CnY6JOdpk0T-waPEowMRTkcMzxcN0GcjPVtLkwH_byw,17196
-anemoi/utils/devtools.py,sha256=vivYRQ1l2aMAkggqhpMJcJedfCrRfpy5XMQjZIsMJS0,3596
+anemoi/utils/devtools.py,sha256=W3OBu96MkXRIl7Qh1SE5Zd6aB1R0QlnmlrlpBYM0fVY,3527
 anemoi/utils/grib.py,sha256=201WcxjjAl92Y2HX2kZ2S8Qr5dN-oG7nV-vQLaybzP4,3610
 anemoi/utils/grids.py,sha256=edTrMK8hpE9ZBzSfwcRftgk0jljNAK3i8CraadILQoM,4427
 anemoi/utils/hindcasts.py,sha256=iYVIxSNFL2HJcc_k1abCFLkpJFGHT8WKRIR4wcAwA3s,2144
-anemoi/utils/humanize.py,sha256=hCrHr5ppREuJR-tBqRqynqe58BHR6Ga_gCQqgEmmrfU,25301
+anemoi/utils/humanize.py,sha256=pjnFJAKHbEAOfcvn8c48kt-8eFy6FGW_U2ruJvfamrA,25189
 anemoi/utils/logs.py,sha256=naTgrmPwWHD4eekFttXftS4gtcAGYHpCqG4iwYprNDA,1804
 anemoi/utils/provenance.py,sha256=tIIgweS0EJyaYzgKwuv3iWny-Gz7N5e5CNWH5MeSYWU,14615
-anemoi/utils/registry.py,sha256=zTnVSCecmrI6SkZhF4ipV7WKiZYHEBJ_5ZwqhRGM4T0,9287
+anemoi/utils/registry.py,sha256=7r14_znm8yr6kv38a7F33MJHM8riRp3tOB_fFF_QQc4,9592
 anemoi/utils/s3.py,sha256=xMT48kbcelcjjqsaU567WI3oZ5eqo88Rlgyx5ECszAU,4074
 anemoi/utils/sanitise.py,sha256=ZYGdSX6qihQANr3pHZjbKnoapnzP1KcrWdW1Ul1mOGk,3668
 anemoi/utils/sanitize.py,sha256=43ZKDcfVpeXSsJ9TFEc9aZnD6oe2cUh151XnDspM98M,462
@@ -30,9 +30,9 @@ anemoi/utils/mars/requests.py,sha256=VFMHBVAAl0_2lOcMBa1lvaKHctN0lDJsI6_U4BucGew
 anemoi/utils/remote/__init__.py,sha256=-uaYFi4yRYFRf46ubQbJo86GCn6HE5VQrcaoyrmyW28,20704
 anemoi/utils/remote/s3.py,sha256=spQ8l0rwQjLZh9dZu5cOsYIvNwKihQfCJ6YsFYegeqI,17339
 anemoi/utils/remote/ssh.py,sha256=xNtsawh8okytCKRehkRCVExbHZj-CRUQNormEHglfuw,8088
-anemoi_utils-0.4.14.dist-info/licenses/LICENSE,sha256=8HznKF1Vi2IvfLsKNE5A2iVyiri3pRjRPvPC9kxs6qk,11354
-anemoi_utils-0.4.14.dist-info/METADATA,sha256=VLuh3lxCAuACOrJRnE_Dm0zX_pVfIHoppmRrP505pkY,15360
-anemoi_utils-0.4.14.dist-info/WHEEL,sha256=1tXe9gY0PYatrMPMDd6jXqjfpz_B-Wqm32CPfRC58XU,91
-anemoi_utils-0.4.14.dist-info/entry_points.txt,sha256=LENOkn88xzFQo-V59AKoA_F_cfYQTJYtrNTtf37YgHY,60
-anemoi_utils-0.4.14.dist-info/top_level.txt,sha256=DYn8VPs-fNwr7fNH9XIBqeXIwiYYd2E2k5-dUFFqUz0,7
-anemoi_utils-0.4.14.dist-info/RECORD,,
+anemoi_utils-0.4.16.dist-info/licenses/LICENSE,sha256=8HznKF1Vi2IvfLsKNE5A2iVyiri3pRjRPvPC9kxs6qk,11354
+anemoi_utils-0.4.16.dist-info/METADATA,sha256=36Oea7pKpe5I3NxOuPP1ILtAm5pGZ2xtvMqsRz-UyvM,15360
+anemoi_utils-0.4.16.dist-info/WHEEL,sha256=1tXe9gY0PYatrMPMDd6jXqjfpz_B-Wqm32CPfRC58XU,91
+anemoi_utils-0.4.16.dist-info/entry_points.txt,sha256=LENOkn88xzFQo-V59AKoA_F_cfYQTJYtrNTtf37YgHY,60
+anemoi_utils-0.4.16.dist-info/top_level.txt,sha256=DYn8VPs-fNwr7fNH9XIBqeXIwiYYd2E2k5-dUFFqUz0,7
+anemoi_utils-0.4.16.dist-info/RECORD,,