add BMI

Author: Mostafa Daoud

Dockerfile

@@ -1,4 +1,4 @@
-FROM ghcr.io/ecoextreml/stemmus_scope:1.5.0
+FROM ghcr.io/ecoextreml/stemmus_scope:1.6.0
 
 LABEL maintainer="Bart Schilperoort <b.schilperoort@esciencecenter.nl>"
 LABEL org.opencontainers.image.source = "https://github.com/EcoExtreML/STEMMUS_SCOPE_Processing"

PyStemmusScope/bmi/implementation.py

@@ -9,7 +9,12 @@
 from bmipy.bmi import Bmi
 from PyStemmusScope.bmi.utils import InapplicableBmiMethods
 from PyStemmusScope.bmi.utils import nested_set
+<<<<<<< HEAD
 from PyStemmusScope.bmi.variable_reference import VARIABLES, BmiVariable
+=======
+from PyStemmusScope.bmi.variable_reference import VARIABLES
+from PyStemmusScope.bmi.variable_reference import BmiVariable
+>>>>>>> main
 from PyStemmusScope.config_io import read_config
 
 
@@ -26,11 +31,19 @@
 MODEL_VARNAMES: tuple[str, ...] = tuple(var.name for var in VARIABLES)
 
 VARNAME_UNITS: dict[str, str] = {var.name: var.units for var in VARIABLES}
+<<<<<<< HEAD
 
 VARNAME_DTYPE: dict[str, str] = {var.name: var.dtype for var in VARIABLES}
 
 VARNAME_GRID: dict[str, int] = {var.name: var.grid for var in VARIABLES}
 
+=======
+
+VARNAME_DTYPE: dict[str, str] = {var.name: var.dtype for var in VARIABLES}
+
+VARNAME_GRID: dict[str, int] = {var.name: var.grid for var in VARIABLES}
+
+>>>>>>> main
 VARNAME_LOC: dict[str, list[str]] = {var.name: var.keys for var in VARIABLES}
 
 NO_STATE_MSG = (
@@ -414,6 +427,9 @@ def set_value(self, name: str, src: np.ndarray) -> None:
         """
         if self.state is None:
             raise ValueError(NO_STATE_MSG)
+        if src.size != self.get_grid_size(self.get_var_grid(name)):
+            msg = f"Size of `src` and variable '{name}' grid size are not equal!"
+            raise ValueError(msg)
         self.state = set_variable(self.state, name, src)
 
     def set_value_at_indices(
@@ -432,6 +448,9 @@ def set_value_at_indices(
         """
         if self.state is None:
             raise ValueError(NO_STATE_MSG)
+        if inds.size != src.size:
+            msg = "Sizes of `inds` and `src` are not equal!"
+            raise ValueError(msg)
         self.state = set_variable(self.state, name, src, inds)
 
     ### GRID INFO ###