Add vulnerability functions.

We add here the vulnerability funcions to use the data from the
ECB's stress test article from 2023.

Co-authored-by: Víctor de Luna <[email protected]>

Signed-off-by: Virginia Morales <[email protected]>

pyproject.toml

@@ -73,10 +73,12 @@ include = ["physrisk*"]
 
 [tool.pdm.dev-dependencies]
 test = [
-  "pdm[pytest]",
-  "pytest",
-  "pytest-cov",
-  "sphinx-pyproject"
+    "pdm[pytest]",
+    "pytest",
+    "pytest-cov",
+    "sphinx-pyproject",
+    "pandas>=2.0.3",
+    "dependency-injector>=4.41.0",
 ]
 lint = [
   "isort",

src/physrisk/api/v1/common.py

@@ -1,3 +1,4 @@
+from enum import Enum
 from typing import Dict, List, Optional, Union
 
 import numpy as np
@@ -23,6 +24,14 @@ class Array(np.ndarray, metaclass=ArrayMeta):
     pass
 
 
+class ModelKind(Enum):
+    """Enumeration to select the kind of vulnerability model and risk model for calculations."""
+
+    DEFAULT = 0
+    STRESS_TEST = 1
+    GENERIC = 2
+
+
 class Asset(BaseModel, extra="allow"):
     """Defines an asset. An asset is identified first by its asset_class and then by its type within the class.
     An asset's value may be impacted through damage or through disruption

src/physrisk/api/v1/impact_req_resp.py

@@ -10,6 +10,7 @@
     VulnerabilityDistrib,
 )
 from physrisk.api.v1.hazard_data import Scenario
+from physrisk.api.v1.common import ModelKind
 
 
 class CalcSettings(BaseModel):
@@ -36,9 +37,9 @@ class AssetImpactRequest(BaseModel):
     include_calc_details: bool = Field(
         True, description="If true, include impact calculation details."
     )
-    use_case_id: str = Field(
-        "",
-        description="Identifier for 'use case' used in the risk measures calculation.",
+    model_kind: ModelKind = Field(
+        ModelKind.DEFAULT,
+        description="Enumeration to select the type of vulnerability model and risk model for calculations. Options: Default, StressTest, Generic",
     )
     provider_max_requests: Dict[str, int] = Field(
         {},
@@ -70,6 +71,8 @@ class Category(int, Enum):
     HIGH = 3
     REDFLAG = 4
 
+    NORISK = -1
+
 
 class RiskMeasureDefinition(BaseModel):
     measure_id: str = Field(None, description="Identifier for the risk measure.")

src/physrisk/container.py

@@ -2,18 +2,14 @@
 
 from dependency_injector import containers, providers
 
+from physrisk.kernel import calculation as calc
 from physrisk.data.hazard_data_provider import SourcePath
 from physrisk.data.inventory import EmbeddedInventory
 from physrisk.data.inventory_reader import InventoryReader
 from physrisk.data.pregenerated_hazard_model import ZarrHazardModel
 from physrisk.data.zarr_reader import ZarrReader
-from physrisk.kernel import calculation as calc
 from physrisk.kernel.hazard_model import HazardModelFactory
-from physrisk.kernel.vulnerability_model import (
-    DictBasedVulnerabilityModels,
-    VulnerabilityModels,
-    VulnerabilityModelsFactory,
-)
+from physrisk.kernel.vulnerability_model import DictBasedVulnerabilityModelsFactory
 from physrisk.requests import Requester, _create_inventory, create_source_paths
 
 
@@ -41,11 +37,6 @@ def hazard_model(
         )
 
 
-class DictBasedVulnerabilityModelsFactory(VulnerabilityModelsFactory):
-    def vulnerability_models(self) -> VulnerabilityModels:
-        return DictBasedVulnerabilityModels(calc.get_default_vulnerability_models())
-
-
 class Container(containers.DeclarativeContainer):
     config = providers.Configuration(default={"zarr_sources": ["embedded", "hazard"]})
 

src/physrisk/data/inventory.py

@@ -2,18 +2,16 @@
 import hashlib
 import importlib.resources
 import json
-import logging
 from collections import defaultdict
 from typing import DefaultDict, Dict, Iterable, List, Tuple
 
-from pydantic import TypeAdapter, parse_obj_as
+from pydantic import TypeAdapter
 
 import physrisk.data.colormap_provider as colormap_provider
 import physrisk.data.static.hazard
+from physrisk.api.v1.hazard_data import HazardResource, Period
 from physrisk.data.inventory_reader import HazardModels
 
-from ..api.v1.hazard_data import HazardResource, Period
-
 # from physrisk.kernel.hazards import ChronicHeat