climate-ref-pmp 0.8.0__py3-none-any.whl → 0.9.0__py3-none-any.whl

climate_ref_pmp/__init__.py

@@ -6,11 +6,18 @@ from __future__ import annotations
 
 import importlib.metadata
 import os
+from pathlib import Path
 from typing import TYPE_CHECKING
 
+import pooch
 from loguru import logger
 
-from climate_ref_core.dataset_registry import DATASET_URL, dataset_registry_manager
+from climate_ref_core.dataset_registry import (
+    DATASET_URL,
+    dataset_registry_manager,
+    fetch_all_files,
+    validate_registry_cache,
+)
 from climate_ref_core.providers import CondaDiagnosticProvider
 from climate_ref_pmp.diagnostics import ENSO, AnnualCycle, ExtratropicalModesOfVariability
 
@@ -19,6 +26,8 @@ if TYPE_CHECKING:
 
 __version__ = importlib.metadata.version("climate-ref-pmp")
 
+_REGISTRY_NAME = "pmp-climatology"
+
 
 # Create the PMP diagnostics provider
 # PMP uses a conda environment to run the diagnostics
@@ -38,6 +47,34 @@ class PMPDiagnosticProvider(CondaDiagnosticProvider):
             logger.debug("Setting env variable 'FI_PROVIDER=tcp'")
             self.env_vars["FI_PROVIDER"] = "tcp"
 
+    def fetch_data(self, config: Config) -> None:
+        """Fetch PMP climatology data."""
+        registry = dataset_registry_manager[_REGISTRY_NAME]
+        fetch_all_files(registry, _REGISTRY_NAME, output_dir=None)
+
+    def validate_setup(self, config: Config) -> bool:
+        """Validate conda environment and data checksums."""
+        # First check conda environment
+        if not super().validate_setup(config):
+            return False
+
+        # Then check data checksums
+        registry = dataset_registry_manager[_REGISTRY_NAME]
+        errors = validate_registry_cache(registry, _REGISTRY_NAME)
+        if errors:
+            for error in errors:
+                logger.error(f"{self.slug} validation failed: {error}")
+            logger.error(
+                f"Data for {self.slug} is missing or corrupted. "
+                f"Please run `ref providers setup --provider {self.slug}` to fetch data."
+            )
+            return False
+        return True
+
+    def get_data_path(self) -> Path | None:
+        """Get the path where PMP data is cached."""
+        return Path(pooch.os_cache("climate_ref"))
+
 
 provider = PMPDiagnosticProvider("PMP", __version__)
 

{climate_ref_pmp-0.8.0.dist-info → climate_ref_pmp-0.9.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: climate-ref-pmp
-Version: 0.8.0
+Version: 0.9.0
 Summary: PMP diagnostic provider for the Rapid Evaluation Framework
 Author-email: Jiwoo Lee <jwlee@llnl.gov>, Jared Lewis <jared.lewis@climate-resource.com>
 License-Expression: Apache-2.0

{climate_ref_pmp-0.8.0.dist-info → climate_ref_pmp-0.9.0.dist-info}/RECORD

@@ -1,4 +1,4 @@
-climate_ref_pmp/__init__.py,sha256=RZ2pZS50d8FctDREpJheBHwX5Y2vZCWtAMgY2bseue4,2267
+climate_ref_pmp/__init__.py,sha256=cSbtESJGEPRLpkPbdBZ4n8SV5xQDMgqETPAC9R1h0uI,3530
 climate_ref_pmp/pmp_driver.py,sha256=ZNuwg7qmx43DlVoW52LY4SUiKV4jxnAroDowVUrrYps,8136
 climate_ref_pmp/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 climate_ref_pmp/dataset_registry/pmp_climatology.txt,sha256=lruldzyr8fS79Tdg5RsIo_nzPGaQDjwpUE00t9S2QTM,3488
@@ -13,9 +13,9 @@ climate_ref_pmp/params/pmp_param_annualcycle_1-clims.py,sha256=P7DkL5wdtK9huQ9um
 climate_ref_pmp/params/pmp_param_annualcycle_2-metrics.py,sha256=JH9flUE9Ti-2bhCpDkDtmup1aD_7brHPqNSbc2-loF8,1655
 climate_ref_pmp/requirements/conda-lock.yml,sha256=HIFVM0uqIBDBmeFqsk35bvF_WkWnMoUKgpryewod0sA,337159
 climate_ref_pmp/requirements/environment.yml,sha256=QUDTJ3eJVcAjUhaGMhwpFWYwKCByyUz0MDEs881Sl-I,157
-climate_ref_pmp-0.8.0.dist-info/METADATA,sha256=6EqPg4E6mGwUG1oUcrcvZdzXq0SmGHWHlXpGMCZTW7o,2733
-climate_ref_pmp-0.8.0.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-climate_ref_pmp-0.8.0.dist-info/entry_points.txt,sha256=zfT0ogvlMMR23ZaS-e7HYiTki0_sUcSh8gGTu4p81kg,55
-climate_ref_pmp-0.8.0.dist-info/licenses/LICENCE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-climate_ref_pmp-0.8.0.dist-info/licenses/NOTICE,sha256=4qTlax9aX2-mswYJuVrLqJ9jK1IkN5kSBqfVvYLF3Ws,128
-climate_ref_pmp-0.8.0.dist-info/RECORD,,
+climate_ref_pmp-0.9.0.dist-info/METADATA,sha256=Bqs_RlqtYh9P4Bt88hVgFwklFFKq99icPKwjHtFxnrI,2733
+climate_ref_pmp-0.9.0.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+climate_ref_pmp-0.9.0.dist-info/entry_points.txt,sha256=zfT0ogvlMMR23ZaS-e7HYiTki0_sUcSh8gGTu4p81kg,55
+climate_ref_pmp-0.9.0.dist-info/licenses/LICENCE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+climate_ref_pmp-0.9.0.dist-info/licenses/NOTICE,sha256=4qTlax9aX2-mswYJuVrLqJ9jK1IkN5kSBqfVvYLF3Ws,128
+climate_ref_pmp-0.9.0.dist-info/RECORD,,