update dependencies and fix import for tomllib; adjust logging for da…

…taset dimensions
openclimatefix · Feb 28, 2025 · 83279dc · 83279dc
1 parent 6cedbda
commit 83279dc
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -42,6 +42,7 @@ dependencies = [
     "h5netcdf",
     "h5py",
     "pvlive-api",
+    "pytest-mock",
 ]
 
 [project.optional-dependencies]
@@ -55,7 +56,6 @@ dev = [
     "twine",
     "pytest",
     "pytest-cov",
-    "pytest-mock",
     "tomli",
 ]
 

diff --git a/src/open_data_pvnet/utils/data_downloader.py b/src/open_data_pvnet/utils/data_downloader.py
@@ -42,7 +42,7 @@ def merge_datasets(datasets: List[xr.Dataset]) -> xr.Dataset:
     ds = xr.merge(datasets, compat="override")
     logger.info("Dataset info:")
     logger.info(f"Variables: {list(ds.variables)}")
-    logger.info(f"Dimensions: {dict(ds.dims)}")
+    logger.info(f"Dimensions: {dict(ds.sizes)}")
     logger.info(f"Coordinates: {list(ds.coords)}")
     return ds
 

diff --git a/tests/test_metadata.py b/tests/test_metadata.py
@@ -1,4 +1,4 @@
-import tomli
+import tomllib
 import open_data_pvnet
 
 
@@ -8,7 +8,7 @@ def test_version_consistency():
     """
     # Read version from pyproject.toml
     with open("pyproject.toml", "rb") as f:
-        pyproject_data = tomli.load(f)
+        pyproject_data = tomllib.load(f)
     pyproject_version = pyproject_data["project"]["version"]
 
     # Read version from the __init__.py file