anemoi-utils 0.3.5__tar.gz → 0.3.6__tar.gz

{anemoi_utils-0.3.5/src/anemoi_utils.egg-info → anemoi_utils-0.3.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: anemoi-utils
-Version: 0.3.5
+Version: 0.3.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_utils-0.3.5 → anemoi_utils-0.3.6}/src/anemoi/utils/_version.py

@@ -12,5 +12,5 @@ __version__: str
 __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
 
-__version__ = version = '0.3.5'
-__version_tuple__ = version_tuple = (0, 3, 5)
+__version__ = version = '0.3.6'
+__version_tuple__ = version_tuple = (0, 3, 6)

{anemoi_utils-0.3.5 → anemoi_utils-0.3.6}/src/anemoi/utils/config.py

@@ -6,10 +6,13 @@
 # nor does it submit to any jurisdiction.
 
 
+import json
 import logging
 import os
 import threading
 
+import yaml
+
 try:
     import tomllib  # Only available since 3.11
 except ImportError:
@@ -41,10 +44,49 @@ class DotDict(dict):
 
     def __init__(self, *args, **kwargs):
         super().__init__(*args, **kwargs)
+
         for k, v in self.items():
             if isinstance(v, dict):
                 self[k] = DotDict(v)
 
+            if isinstance(v, list):
+                self[k] = [DotDict(i) if isinstance(i, dict) else i for i in v]
+
+            if isinstance(v, tuple):
+                self[k] = [DotDict(i) if isinstance(i, dict) else i for i in v]
+
+    @classmethod
+    def from_file(cls, path: str):
+        _, ext = os.path.splitext(path)
+        if ext == ".yaml" or ext == ".yml":
+            return cls.from_yaml_file(path)
+        elif ext == ".json":
+            return cls.from_json_file(path)
+        elif ext == ".toml":
+            return cls.from_toml_file(path)
+        else:
+            raise ValueError(f"Unknown file extension {ext}")
+
+    @classmethod
+    def from_yaml_file(cls, path: str):
+        with open(path, "r") as file:
+            data = yaml.safe_load(file)
+
+        return cls(data)
+
+    @classmethod
+    def from_json_file(cls, path: str):
+        with open(path, "r") as file:
+            data = json.load(file)
+
+        return cls(data)
+
+    @classmethod
+    def from_toml_file(cls, path: str):
+        with open(path, "r") as file:
+            data = tomllib.load(file)
+        return cls(data)
+
     def __getattr__(self, attr):
         try:
             return self[attr]

{anemoi_utils-0.3.5 → anemoi_utils-0.3.6/src/anemoi_utils.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: anemoi-utils
-Version: 0.3.5
+Version: 0.3.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_utils-0.3.5 → anemoi_utils-0.3.6}/tests/test_utils.py

@@ -12,7 +12,7 @@ from anemoi.utils.grib import shortname_to_paramid
 
 
 def test_dotdict():
-    d = DotDict(a=1, b=2, c=dict(d=3, e=4))
+    d = DotDict(a=1, b=2, c=dict(d=3, e=4), e=[1, dict(a=3), 3])
     assert d.a == 1
     assert d.b == 2
     assert d.c.d == 3
@@ -27,6 +27,8 @@ def test_dotdict():
     d.d.x = 6
     assert d.d.x == 6
 
+    assert d.e[1].a == 3
+
 
 def test_grib():
     assert shortname_to_paramid("2t") == 167